tf.contrib.eager.run_test_in_graph_and_eager_modes

Выполнить помеченный тест с включенной и выключенной функцией eager выполнения.

tf.contrib.eager.run_test_in_graph_and_eager_modes(
    func=None, config=None, use_gpu=True, reset_test=True,
    assert_no_eager_garbage=False
)

Эта функция возвращает декоратор, предназначенный для применения к методам тестов в классе tf.test.TestCase. Это заставит содержимое метода теста выполняться дважды — один раз в обычном режиме и один раз с включённым eager выполнением. Это позволяет юнит-тестам подтвердить эквивалентность выполнения в режиме eager и в режиме графа (см. tf.compat.v1.enable_eager_execution).

Например, рассмотрим следующий юнит-тест:

class MyTests(tf.test.TestCase):

  @run_in_graph_and_eager_modes
  def test_foo(self):
    x = tf.constant([1, 2])
    y = tf.constant([3, 4])
    z = tf.add(x, y)
    self.assertAllEqual([4, 6], self.evaluate(z))

if __name__ == "__main__":
  tf.test.main()

Этот тест проверяет, что tf.add() имеет одинаковое поведение при вычислении с включённым eager выполнением, что и при построении графика TensorFlow и выполнении z тензора в сессии.

deprecated_graph_mode_only, run_v1_only, run_v2_only, и run_in_graph_and_eager_modes являются доступными декораторами для различных комбинаций v1/v2/eager/graph.