Disable certain python tests for TFRT.

These tests target features that TFRT don't have plan to support for single host training, including quantization, XLA, clusters.

PiperOrigin-RevId: 343969137
Change-Id: Ib0d3daa1f3b38545a30030b4be321ca40d15062e

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",

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 = [

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)

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 = [

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 = [

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({