Internal change.

PiperOrigin-RevId: 236374895
2019-03-01 14:06:45 -08:00 · 2019-03-01 14:06:45 -08:00 · 234a8d2440
commit 234a8d2440
parent 9dc8d7323e
1 changed files with 1 additions and 0 deletions
--- a/tensorflow/python/kernel_tests/BUILD
+++ b/tensorflow/python/kernel_tests/BUILD
@ -2630,6 +2630,7 @@ cuda_py_test(
        "//tensorflow/python/eager:context",
    ],
    flaky = 1,  # create_local_cluster sometimes times out.
+    shard_count = 10,
    tags = ["no_rocm"],
    xla_enable_strict_auto_jit = True,
 )