Increase shard count for keras_embedding_model_correctness_test.

PiperOrigin-RevId: 327538956
Change-Id: I1cc98e6e40e507500758230341440d90bc6d8d85
This commit is contained in:
A. Unique TensorFlower 2020-08-19 17:20:54 -07:00 committed by TensorFlower Gardener
parent e2ff54f938
commit 8257866ced

View File

@ -417,7 +417,7 @@ distribute_py_test(
srcs = ["keras_embedding_model_correctness_test.py"],
full_precision = True,
main = "keras_embedding_model_correctness_test.py",
shard_count = 4,
shard_count = 8,
tags = [
"multi_and_single_gpu",
"no_windows_gpu",