Disable keras layer_correctness_test for TFRT.

PiperOrigin-RevId: 357002490 Change-Id: I28966d39de769c7388a0ef300d2006c070b20c14
2021-02-11 10:14:29 -08:00 · 2021-02-11 10:14:29 -08:00 · d73cfcdbba
commit d73cfcdbba
parent 2d7463e97e
1 changed files with 4 additions and 1 deletions
--- a/tensorflow/python/keras/mixed_precision/BUILD
+++ b/tensorflow/python/keras/mixed_precision/BUILD
@ -289,7 +289,10 @@ cuda_py_test(
    srcs = ["layer_correctness_test.py"],
    python_version = "PY3",
    shard_count = 10,
-    tags = ["no_rocm"],
+    tags = [
+        "no_rocm",
+        "no_tfrt",  # TODO(b/179863362)
+    ],
    deps = [
        "//tensorflow/python:client_testlib",
        "//tensorflow/python/compat:v2_compat",