Shard a slow test

PiperOrigin-RevId: 310904207 Change-Id: Ic3fd8eb46d70482ca283c7bed6869274d8353acc
2020-05-11 07:02:46 -07:00 · 2020-05-11 07:02:46 -07:00 · 3ba00b04c1
parent 3354e9eb51
commit 3ba00b04c1
1 changed files with 1 additions and 0 deletions
--- a/tensorflow/compiler/tests/BUILD
+++ b/tensorflow/compiler/tests/BUILD
@ -1386,6 +1386,7 @@ tf_xla_py_test(
    size = "medium",
    srcs = ["fused_batchnorm_test.py"],
    python_version = "PY3",
+    shard_count = 5,
    tags = [
        "no_pip",  # TODO(b/149738646): fix pip install so these tests run on kokoro pip
    ],