diff options
Diffstat (limited to 'tensorflow/python/kernel_tests/conv2d_transpose_test.py')
-rw-r--r-- | tensorflow/python/kernel_tests/conv2d_transpose_test.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/tensorflow/python/kernel_tests/conv2d_transpose_test.py b/tensorflow/python/kernel_tests/conv2d_transpose_test.py index 77f783d915..d260660e10 100644 --- a/tensorflow/python/kernel_tests/conv2d_transpose_test.py +++ b/tensorflow/python/kernel_tests/conv2d_transpose_test.py @@ -159,8 +159,8 @@ class Conv2DTransposeTest(tf.test.TestCase): self.assertLess(err, err_tolerance) def testConv2DTransposeSingleStrideNCHW(self): - # `NCHW` data fomat is only supported for `GPU` device. - if tf.test.is_gpu_available(): + # `NCHW` data fomat is only supported for CUDA device. + if tf.test.is_gpu_available(cuda_only=True): with self.test_session(use_gpu=True): strides = [1, 1, 1, 1] @@ -192,8 +192,8 @@ class Conv2DTransposeTest(tf.test.TestCase): self.assertAllClose(target, value[n, k, h, w]) def testConv2DTransposeSameNCHW(self): - # `NCHW` data fomat is only supported for `GPU` device. - if tf.test.is_gpu_available(): + # `NCHW` data fomat is only supported for CUDA device. + if tf.test.is_gpu_available(cuda_only=True): with self.test_session(use_gpu=True): strides = [1, 1, 2, 2] @@ -226,8 +226,8 @@ class Conv2DTransposeTest(tf.test.TestCase): self.assertAllClose(target, value[n, k, h, w]) def testConv2DTransposeValidNCHW(self): - # `NCHW` data fomat is only supported for `GPU` device. - if tf.test.is_gpu_available(): + # `NCHW` data fomat is only supported for CUDA device. + if tf.test.is_gpu_available(cuda_only=True): with self.test_session(use_gpu=True): strides = [1, 1, 2, 2] |