From e87fbf3b6c5e36b4326249cfea4655b27e40cf07 Mon Sep 17 00:00:00 2001 From: Yanhui Liang Date: Fri, 17 Jan 2020 22:34:12 -0800 Subject: [PATCH] Enable multi_worker_test with "multi_and_single_gpu" tag. PiperOrigin-RevId: 290397331 Change-Id: Id88083fff97a0bed45829952d10812cd0ddfa91b --- tensorflow/python/keras/distribute/BUILD | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/python/keras/distribute/BUILD b/tensorflow/python/keras/distribute/BUILD index ac98153c17d..5dce088b3dd 100644 --- 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",