def METHOD_NAME(self):
    """ Test get output tensor utility """
    AimetLogger.set_level_for_all_areas(logging.DEBUG)
    tf.compat.v1.reset_default_graph()
    graph = tf.Graph()
    with graph.as_default():
        _ = keras_model()
    op = graph.get_operation_by_name('conv2d/Conv2D')
    output_tensor = ActivationSampler._get_output_tensor(op)
    self.assertEqual(output_tensor.op.type, 'BiasAdd')
    self.assertNotEqual(output_tensor.op.type, 'Conv2D')