diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-12 09:09:44 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-12 09:15:05 -0700 |
commit | 9e78991b5c380b7fba0444685e5c6ef40e3c5b26 (patch) | |
tree | 27190875b6ba0929b95e46929fab3a5208e8aa96 /tensorflow/python/kernel_tests/control_flow_ops_py_test.py | |
parent | 6995d2b9be0e398f11a17348eb5b4745aee0af0d (diff) |
Fix typo in Tensorflow control_flow_ops_py_test.
The test would fall back to GPU:0 when unable to find a GPU. This should be CPU.
PiperOrigin-RevId: 212649435
Diffstat (limited to 'tensorflow/python/kernel_tests/control_flow_ops_py_test.py')
-rw-r--r-- | tensorflow/python/kernel_tests/control_flow_ops_py_test.py | 2 |
1 files changed, 1 insertions, 1 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 bdf7e0e4a0..a03e217ddc 100644 --- a/tensorflow/python/kernel_tests/control_flow_ops_py_test.py +++ b/tensorflow/python/kernel_tests/control_flow_ops_py_test.py @@ -1753,7 +1753,7 @@ class ControlFlowTest(test.TestCase): def _testWhileGrad_ColocateGradients(self, colocate): gpu_dev_name = test.gpu_device_name() if test.is_gpu_available( - ) else "/device:GPU:0" + ) else "/device:CPU:0" graph = ops.Graph() with graph.as_default(): |