Increase number of shards to prevent flaky timeouts

PiperOrigin-RevId: 321371211 Change-Id: I3508bd5ffc367edec5bad81c19f2ae4219841c47
2020-07-15 09:05:30 -07:00 · 2020-07-15 09:05:30 -07:00 · c9db66374c
commit c9db66374c
parent de473e40fc
1 changed files with 1 additions and 1 deletions
--- a/tensorflow/python/kernel_tests/random/BUILD
+++ b/tensorflow/python/kernel_tests/random/BUILD
@ -119,7 +119,7 @@ cuda_py_test(
    name = "stateless_random_ops_test",
    size = "medium",
    srcs = ["stateless_random_ops_test.py"],
-    shard_count = 5,
+    shard_count = 10,
    tfrt_enabled = True,
    deps = [
        "//tensorflow/python:array_ops",