experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Enable critical_section_test with control flow v2.

Browse Source 
PiperOrigin-RevId: 252982317
This commit is contained in:
Saurabh Saxena 2019-06-13 01:16:02 -07:00 committed by TensorFlower Gardener
parent 0737c7abc7
commit ad27775640
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/python/kernel_tests/critical_section_test.py
View File

@ -34,6 +34,7 @@ from tensorflow.python.platform import tf_logging as logging
# from tensorflow.python.training import saver as saver_lib # from tensorflow.python.training import saver as saver_lib
@test_util.with_control_flow_v2
class CriticalSectionTest(test.TestCase): class CriticalSectionTest(test.TestCase):
@test_util.run_in_graph_and_eager_modes @test_util.run_in_graph_and_eager_modes
@ -55,6 +56,8 @@ class CriticalSectionTest(test.TestCase):
self.assertAllClose([2.0 * i for i in range(num_concurrent)], self.assertAllClose([2.0 * i for i in range(num_concurrent)],
sorted(r_value)) sorted(r_value))
@test_util.disable_control_flow_v2("b/135070612")
@test_util.run_v1_only("b/135070612")
@test_util.run_in_graph_and_eager_modes @test_util.run_in_graph_and_eager_modes
@test_util.xla_allow_fallback("b/128495870") @test_util.xla_allow_fallback("b/128495870")
def testCriticalSectionWithControlFlow(self): def testCriticalSectionWithControlFlow(self):