tf.test.with_eager_op_as_function

Добавляет методы, которые вызывают исходные методы с включённым eager_op_as_function.

tf.test.with_eager_op_as_function(
    cls=None, only_as_function=False
)

Пример:

@test_util.with_eager_op_as_function class SessionTest(test.TestCase):

def testEnabledForEagerOpAsFunction(self): ...

@disable_eager_op_as_function("b/xyzabc") def testDisabledForEagerOpAsFunction(self): ...

Сгенерированный класс:

class SessionTest(test.TestCase):

def testEnabledForEagerOpAsFunction(self): ...

def testEnabledForEagerOpAsFunctionWithEagerOpAsFunctionEnabled(self): // Включить run_eager_op_as_function // Сбросить контекст testEnabledForEagerOpAsFunction(self) // Отключить run_eager_op_as_function // Сбросить контекст

def testDisabledForEagerOpAsFunction(self): ...