Disable tests for CUDA 11 that fail CI.

PiperOrigin-RevId: 319277399
Change-Id: I437121a8b44ea1da717eb2452a768103908c51ca

tensorflow/python/keras/distribute/BUILD

@@ -120,6 +120,7 @@ distribute_py_test(
     main = "custom_training_loop_models_test.py",
     tags = [
         "multi_and_single_gpu",
+        "no_cuda11",
     ],
     tpu_tags = [
         "no_oss",  # b/153615544.

tensorflow/python/keras/layers/BUILD

@@ -519,6 +519,7 @@ cuda_py_test(
     python_version = "PY3",
     shard_count = 4,
     tags = [
+        "no_cuda11",
         "no_windows_gpu",
     ],
     deps = [
@@ -784,6 +785,7 @@ cuda_py_test(
     srcs = ["lstm_v2_test.py"],
     python_version = "PY3",
     shard_count = 12,
+    tags = ["no_cuda11"],
     deps = [
         "//tensorflow/python:client_testlib",
         "//tensorflow/python/keras",
@@ -798,6 +800,7 @@ cuda_py_test(
     srcs = ["gru_v2_test.py"],
     python_version = "PY3",
     shard_count = 12,
+    tags = ["no_cuda11"],
     deps = [
         "//tensorflow/python:client_testlib",
         "//tensorflow/python/keras",

tensorflow/python/keras/tests/BUILD

@@ -274,6 +274,7 @@ cuda_py_test(
     name = "op_callbacks_test",
     srcs = ["op_callbacks_test.py"],
     python_version = "PY3",
+    tags = ["no_cuda11"],
     deps = [
         "//tensorflow/python:framework_ops",
         "//tensorflow/python:framework_test_lib",