diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD index 31c205c638e..106b21b342d 100644 --- a/tensorflow/python/BUILD +++ b/tensorflow/python/BUILD @@ -6841,7 +6841,10 @@ tf_py_test( size = "small", srcs = ["ops/dequantize_op_test.py"], python_version = "PY3", - tags = ["no_windows"], + tags = [ + "no_tfrt", # TODO(b/169901260) + "no_windows", + ], deps = [ ":array_ops", ":client_testlib", @@ -6855,7 +6858,10 @@ tf_py_test( size = "small", srcs = ["ops/quantized_ops_test.py"], python_version = "PY3", - tags = ["no_windows"], + tags = [ + "no_tfrt", # TODO(b/169901260) + "no_windows", + ], deps = [ ":array_ops", ":client_testlib", @@ -6869,7 +6875,10 @@ tf_py_test( size = "small", srcs = ["ops/quantized_conv_ops_test.py"], python_version = "PY3", - tags = ["no_windows"], + tags = [ + "no_tfrt", # TODO(b/169901260) + "no_windows", + ], deps = [ ":client_testlib", ":framework_for_generated_wrappers", diff --git a/tensorflow/python/eager/BUILD b/tensorflow/python/eager/BUILD index 89fdc0c82ad..e48a02ab4d8 100644 --- a/tensorflow/python/eager/BUILD +++ b/tensorflow/python/eager/BUILD @@ -1047,6 +1047,7 @@ cuda_py_test( shard_count = 8, tags = [ "no_oss", # This test launches local server + "no_tfrt", # TODO(b/171765113) ], deps = [ "//tensorflow/python:array_ops", @@ -1067,6 +1068,7 @@ cuda_py_test( shard_count = 8, tags = [ "no_oss", # This test launches local server + "no_tfrt", # TODO(b/171765113) "notsan", # TODO(b/170783249) ], deps = [ diff --git a/tensorflow/python/keras/integration_test/BUILD b/tensorflow/python/keras/integration_test/BUILD index 1190d7897f2..f9a98626832 100644 --- a/tensorflow/python/keras/integration_test/BUILD +++ b/tensorflow/python/keras/integration_test/BUILD @@ -111,6 +111,7 @@ tf_py_test( python_version = "PY3", shard_count = 3, tags = [ + "no_tfrt", # TODO(b/171765113) "noasan", # TODO(b/156029134) "nomsan", # TODO(b/156029134) "notsan", # TODO(b/156029134) diff --git a/tensorflow/python/kernel_tests/BUILD b/tensorflow/python/kernel_tests/BUILD index 8fa5e2b33f0..3dac22c413a 100644 --- a/tensorflow/python/kernel_tests/BUILD +++ b/tensorflow/python/kernel_tests/BUILD @@ -277,6 +277,7 @@ cuda_py_test( srcs = ["collective_ops_test.py"], tags = [ "multi_and_single_gpu", + "no_tfrt", # TODO(b/171765113) ], deps = [ "//tensorflow/python:array_ops", @@ -302,6 +303,7 @@ tf_py_test( python_version = "PY3", tags = [ "no_rocm", + "no_tfrt", # TODO(b/171765113) "notsan", # TODO(b/171435192) ], deps = [ diff --git a/tensorflow/python/training/saving/BUILD b/tensorflow/python/training/saving/BUILD index 7a65e3a77e8..218ce9dd0b1 100644 --- a/tensorflow/python/training/saving/BUILD +++ b/tensorflow/python/training/saving/BUILD @@ -40,6 +40,7 @@ cuda_py_test( "functional_saver_test.py", ], tags = [ + "no_tfrt", # TODO(b/171765113) "no_windows", # TODO(b/171350346) ], deps = [ diff --git a/tensorflow/python/training/saving/functional_saver_test.py b/tensorflow/python/training/saving/functional_saver_test.py index 8f3eef4fb9c..e2c9e7c5825 100644 --- a/tensorflow/python/training/saving/functional_saver_test.py +++ b/tensorflow/python/training/saving/functional_saver_test.py @@ -128,6 +128,7 @@ class SaverTest(test.TestCase): second_saver.restore(save_path) self.assertEqual(2., self.evaluate(v2)) + @test_util.disable_tfrt("b/171765113: server is not supported in TFRT yet.") def test_checkpoint_is_sharded_by_task(self): servers = [server_lib.Server.create_local_server() for _ in range(3)] cluster_spec = server_lib.ClusterSpec({