diff options
Diffstat (limited to 'tensorflow/python/kernel_tests/BUILD')
| -rw-r--r-- | tensorflow/python/kernel_tests/BUILD | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/tensorflow/python/kernel_tests/BUILD b/tensorflow/python/kernel_tests/BUILD index 8a6614c837..adf97569ab 100644 --- a/tensorflow/python/kernel_tests/BUILD +++ b/tensorflow/python/kernel_tests/BUILD @@ -1525,6 +1525,7 @@ cuda_py_test( "//tensorflow/python:framework_for_generated_wrappers", "//tensorflow/python:math_ops", ], + tags = ["no_windows_gpu"], ) cuda_py_test( @@ -2057,6 +2058,7 @@ cuda_py_test( "//tensorflow/python:framework_for_generated_wrappers", "//tensorflow/python:math_ops", ], + tags = ["no_windows_gpu"], ) tf_py_test( @@ -2755,6 +2757,7 @@ cuda_py_test( "//tensorflow/python:embedding_ops", "//tensorflow/python:framework", "//tensorflow/python:framework_for_generated_wrappers", + "//tensorflow/python:init_ops", "//tensorflow/python:linalg_ops", "//tensorflow/python:math_ops", "//tensorflow/python:partitioned_variables", @@ -3091,7 +3094,7 @@ tf_py_test( tf_py_test( name = "cond_v2_test", - size = "small", + size = "medium", srcs = ["cond_v2_test.py"], additional_deps = [ "//tensorflow/python:array_ops", @@ -3106,4 +3109,5 @@ tf_py_test( "//tensorflow/python:training", ], grpc_enabled = True, + tags = ["no_gpu"], # TODO(b/111656070) ) |