diff options
| -rw-r--r-- | tensorflow/python/kernel_tests/control_flow_ops_py_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/kernel_tests/control_flow_ops_py_test.py b/tensorflow/python/kernel_tests/control_flow_ops_py_test.py index baea5c0f6d..a5f85b97f7 100644 --- a/tensorflow/python/kernel_tests/control_flow_ops_py_test.py +++ b/tensorflow/python/kernel_tests/control_flow_ops_py_test.py @@ -1116,8 +1116,8 @@ class ControlFlowTest(test.TestCase): self.assertAllClose(10.0, r.eval()) def testWhile_Gpu_2(self): - self._testWhile_Gpu_1(use_gpu=False) - self._testWhile_Gpu_1(use_gpu=True) + self._testWhile_Gpu_2(use_gpu=False) + self._testWhile_Gpu_2(use_gpu=True) def testWhileShape(self): with self.cached_session(): |