From 234a8d244044ea7a14b7eca85b39e7db3641a467 Mon Sep 17 00:00:00 2001
From: Jeremy Lau <lauj@google.com>
Date: Fri, 1 Mar 2019 14:06:45 -0800
Subject: [PATCH] Internal change.

PiperOrigin-RevId: 236374895
---
 tensorflow/python/kernel_tests/BUILD | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tensorflow/python/kernel_tests/BUILD b/tensorflow/python/kernel_tests/BUILD
index 3ab268d7318..7f489cd330e 100644
--- 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,
 )