diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-28 13:31:01 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-28 14:28:09 -0700 |
| commit | 19143aa0e2ac3cdf0d6826e7e1d00cd864080394 (patch) | |
| tree | 066150dfb4b207d5146475867a825b375214654b /tensorflow/python/kernel_tests | |
| parent | c30e729f8f830ea2da46eaa7a5354395c5119def (diff) | |
Internal change.
PiperOrigin-RevId: 214986756
Diffstat (limited to 'tensorflow/python/kernel_tests')
| -rw-r--r-- | tensorflow/python/kernel_tests/BUILD | 22 |
1 files changed, 15 insertions, 7 deletions
diff --git a/tensorflow/python/kernel_tests/BUILD b/tensorflow/python/kernel_tests/BUILD index 65b9e04ed9..9490746fd9 100644 --- a/tensorflow/python/kernel_tests/BUILD +++ b/tensorflow/python/kernel_tests/BUILD @@ -2359,7 +2359,7 @@ cuda_py_test( cuda_py_test( name = "transpose_op_test", - size = "large", + size = "medium", srcs = ["transpose_op_test.py"], additional_deps = [ "//third_party/py/numpy", @@ -2367,10 +2367,11 @@ cuda_py_test( "//tensorflow/python:client_testlib", "//tensorflow/python:framework_for_generated_wrappers", ], - shard_count = 2, + shard_count = 4, tags = [ "no_gpu", "no_oss", + "optonly", # times out ], ) @@ -2489,6 +2490,7 @@ cuda_py_test( "//tensorflow/python:nn_grad", "//tensorflow/python:nn_ops", ], + shard_count = 2, tags = [ "optonly", # flaky timeouts unless optimized ], @@ -2509,7 +2511,7 @@ cuda_py_test( cuda_py_test( name = "conv_ops_test", - size = "large", + size = "medium", srcs = ["conv_ops_test.py"], additional_deps = [ "//third_party/py/numpy", @@ -2528,6 +2530,9 @@ cuda_py_test( "//tensorflow/python:variables", ], shard_count = 4, + tags = [ + "optonly", # times out + ], ) cuda_py_test( @@ -2587,7 +2592,7 @@ cuda_py_test( cuda_py_test( name = "fft_ops_test", - size = "large", + size = "medium", srcs = ["fft_ops_test.py"], additional_deps = [ "//third_party/py/numpy", @@ -2597,7 +2602,8 @@ cuda_py_test( "//tensorflow/python:spectral_ops", "//tensorflow/python:spectral_ops_test_util", ], - shard_count = 3, + shard_count = 4, + tags = ["optonly"], ) cuda_py_test( @@ -2662,7 +2668,7 @@ cuda_py_test( cuda_py_test( name = "scatter_ops_test", - size = "large", # NOTE: This is not run by default. + size = "medium", # NOTE: This is not run by default. srcs = ["scatter_ops_test.py"], additional_deps = [ "//third_party/py/numpy", @@ -2671,11 +2677,13 @@ cuda_py_test( "//tensorflow/python:state_ops", "//tensorflow/python:variables", ], + shard_count = 2, + tags = ["optonly"], ) cuda_py_test( name = "slice_op_test", - size = "large", + size = "medium", srcs = ["slice_op_test.py"], additional_deps = [ "//third_party/py/numpy", |