diff options
Diffstat (limited to 'tensorflow/python/kernel_tests/conv_ops_test.py')
-rw-r--r-- | tensorflow/python/kernel_tests/conv_ops_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/kernel_tests/conv_ops_test.py b/tensorflow/python/kernel_tests/conv_ops_test.py index 35e0a39840..65fa69bb65 100644 --- a/tensorflow/python/kernel_tests/conv_ops_test.py +++ b/tensorflow/python/kernel_tests/conv_ops_test.py @@ -181,8 +181,8 @@ def GetTestConfigs(): all the valid test configs as tuples of data_format and use_gpu. """ test_configs = [("NHWC", False), ("NHWC", True)] - if test.is_gpu_available(): - # "NCHW" format is not currently supported on CPU. + if test.is_gpu_available(cuda_only=True): + # "NCHW" format is only supported on CUDA. test_configs += [("NCHW", True)] return test_configs |