From 43cb5e8be4e954dba9abe74c472f732b5a7604d6 Mon Sep 17 00:00:00 2001 From: Rohan Jain Date: Thu, 20 Dec 2018 14:08:45 -0800 Subject: [PATCH] Disabling some parallel_for tests that fail in eager mode. PiperOrigin-RevId: 226390986 --- tensorflow/python/ops/parallel_for/control_flow_ops_test.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tensorflow/python/ops/parallel_for/control_flow_ops_test.py b/tensorflow/python/ops/parallel_for/control_flow_ops_test.py index 933bddd8cca..8acb0d839c5 100644 --- a/tensorflow/python/ops/parallel_for/control_flow_ops_test.py +++ b/tensorflow/python/ops/parallel_for/control_flow_ops_test.py @@ -27,6 +27,7 @@ from tensorflow.core.example import example_pb2 from tensorflow.core.example import feature_pb2 from tensorflow.python.client import session from tensorflow.python.eager import backprop +from tensorflow.python.eager import context from tensorflow.python.framework import constant_op from tensorflow.python.framework import dtypes from tensorflow.python.framework import ops @@ -104,6 +105,10 @@ class PForTest(test.TestCase): flags.FLAGS.op_conversion_fallback_to_while_loop = False def test_parallel_iterations(self): + # TODO(b/121334512): Remove this check once this passes in Eager mode. + if context.executing_eagerly(): + return + for parallel_iterations in [2, 3, 8, 10]: x = random_ops.random_uniform([8, 3])