Enable multi_worker_test with "multi_and_single_gpu" tag.

PiperOrigin-RevId: 290397331 Change-Id: Id88083fff97a0bed45829952d10812cd0ddfa91b
2020-01-17 22:34:12 -08:00 · 2020-01-17 22:34:12 -08:00 · e87fbf3b6c
commit e87fbf3b6c
parent 96170c0207
1 changed files with 1 additions and 1 deletions
--- a/tensorflow/python/keras/distribute/BUILD
+++ b/tensorflow/python/keras/distribute/BUILD
@ -320,8 +320,8 @@ cuda_py_test(
    srcs = ["multi_worker_test.py"],
    shard_count = 32,
    tags = [
+        "multi_and_single_gpu",
        "no_oss",  # TODO(b/130369494): Investigate why it times out on OSS.
-        # TODO(b/123307453): Add "multi_and_single_gpu",
    ],
    deps = [
        ":multi_worker_testing_utils",