Increase shard_count of strategy_gather_test to avoid timeout:

PiperOrigin-RevId: 337996150
Change-Id: Ifa68a95b8a21f88c5e7fe9dd5f8796afec37a2e3
This commit is contained in:
Chen Chen 2020-10-19 21:59:42 -07:00 committed by TensorFlower Gardener
parent bb3dda7877
commit d9e09b0723

View File

@ -1760,7 +1760,7 @@ distribute_py_test(
srcs = ["strategy_gather_test.py"],
disable_mlir_bridge = False,
python_version = "PY3",
shard_count = 2,
shard_count = 4,
tags = [
"multi_and_single_gpu",
"notsan", # TODO(b/160006974)