Internal change.

PiperOrigin-RevId: 219229787
2018-10-29 18:30:49 -07:00 · 2018-10-29 18:30:49 -07:00 · 10f259865a
commit 10f259865a
parent aaade421b3
1 changed files with 1 additions and 0 deletions
--- 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",