[SYCL] Disables NCHW target for conv_ops_tests

lukeiwanski · Apr 30, 2018 · 79966b6 · 79966b6
1 parent b45338b
commit 79966b6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow/python/kernel_tests/conv_ops_test.py b/tensorflow/python/kernel_tests/conv_ops_test.py
@@ -149,7 +149,7 @@ 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():
+  if test.is_gpu_available(cuda_only=True):
     # "NCHW" format is only supported on CUDA.
     test_configs += [("NCHW", True)]
   return test_configs