commit | ae5ec1f4dfaa712798ea6786b54bd1f2a6234e9d | [log] [tgz] |
---|---|---|
author | Pavithra Vijay <psv@google.com> | Fri Oct 16 17:22:21 2020 -0700 |
committer | TensorFlower Gardener <gardener@tensorflow.org> | Fri Oct 16 17:36:41 2020 -0700 |
tree | b175cfe83e81b03ae33ad99e860b1960233d90ac | |
parent | 1ce9513e90ccea68421e6bc5bd512ff7f98b3c59 [diff] |
Increase shard count for test that times out PiperOrigin-RevId: 337606016 Change-Id: Ic02ea265aa223b050fa0f49061b2b4cf5ff34aa8
diff --git a/tensorflow/python/eager/BUILD b/tensorflow/python/eager/BUILD index bb56146..300da45 100644 --- a/tensorflow/python/eager/BUILD +++ b/tensorflow/python/eager/BUILD
@@ -1119,6 +1119,7 @@ size = "small", srcs = ["device_placement_test.py"], python_version = "PY3", + shard_count = 5, deps = [ ":context", ":def_function",