From 8cf05f47a5c2c8bf40e6e2ee75267fde7f83d107 Mon Sep 17 00:00:00 2001
From: Shanqing Cai <cais@google.com>
Date: Tue, 3 Dec 2019 14:31:17 -0800
Subject: [PATCH] [tfdbg] Shard distributed_callbacks_test to reduce likelihood
 of timeouts

PiperOrigin-RevId: 283623507
Change-Id: I021d13454667d2e92495a5361195b899cdc6eea9
---
 tensorflow/python/debug/BUILD | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tensorflow/python/debug/BUILD b/tensorflow/python/debug/BUILD
index 1c30328e7dd..97fe48ee165 100644
--- a/tensorflow/python/debug/BUILD
+++ b/tensorflow/python/debug/BUILD
@@ -725,6 +725,7 @@ cuda_py_test(
         "//tensorflow/python/keras",
     ],
     python_version = "PY3",
+    shard_count = 8,
     tags = [
         "guitar",
         "multi_and_single_gpu",