Internal change.

PiperOrigin-RevId: 219229787
adology · Oct 30, 2018 · 10f2598 · 10f2598
1 parent aaade42
commit 10f2598
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tensorflow/contrib/distribute/python/BUILD b/tensorflow/contrib/distribute/python/BUILD
@@ -514,6 +514,7 @@ cuda_py_test(
         "//tensorflow/python:platform",
         "//tensorflow/python:summary",
     ],
+    shard_count = 5,
     tags = [
         "multi_and_single_gpu",
         "no_pip",