From c9db66374c611722683e27520ed29c66e1540359 Mon Sep 17 00:00:00 2001
From: Mihai Maruseac <mihaimaruseac@google.com>
Date: Wed, 15 Jul 2020 09:05:30 -0700
Subject: [PATCH] Increase number of shards to prevent flaky timeouts

PiperOrigin-RevId: 321371211
Change-Id: I3508bd5ffc367edec5bad81c19f2ae4219841c47
---
 tensorflow/python/kernel_tests/random/BUILD | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tensorflow/python/kernel_tests/random/BUILD b/tensorflow/python/kernel_tests/random/BUILD
index f7d031154e3..06360fc2095 100644
--- 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",