message: Disable mlir support for some tests that fail with it.
diff: @@ -335,6 +335,7 @@ distribute_py_test(
  distribute_py_test(
  name = "distribute_strategy_test",
  srcs = ["distribute_strategy_test.py"],
  + disable_mlir_bridge = True, # TODO(b/170352626)
  full_precision = True,
  main = "distribute_strategy_test.py",
  python_version = "PY3",
  @@ -402,7 +403,7 @@ distribute_py_test(
  name = "keras_dnn_correctness_test",
  size = "medium",
  srcs = ["keras_dnn_correctness_test.py"],
  - disable_mlir_bridge = False,
  + disable_mlir_bridge = True, # TODO(b/170352626)
  full_precision = True,
  main = "keras_dnn_correctness_test.py",
  # Shard count is set to an odd number to distribute tasks across
  @@ -423,6 +424,7 @@ distribute_py_test(
  name = "keras_embedding_model_correctness_test",
  size = "medium",
  srcs = ["keras_embedding_model_correctness_test.py"],
  + disable_mlir_bridge = True, # TODO(b/170352626)
  full_precision = True,
  main = "keras_embedding_model_correctness_test.py",
  shard_count = 8,
  @@ -440,7 +442,7 @@ distribute_py_test(
  name = "keras_image_model_correctness_test",
  size = "medium",
  srcs = ["keras_image_model_correctness_test.py"],
  - disable_mlir_bridge = False,
  + disable_mlir_bridge = True, # TODO(b/170352626)
  full_precision = True,
  main = "keras_image_model_correctness_test.py",
  shard_count = 16,
  @@ -553,6 +555,7 @@ distribute_py_test(
  distribute_py_test(
  name = "keras_utils_test",
  srcs = ["keras_utils_test.py"],
  + disable_mlir_bridge = True, # TODO(b/170352626)
  full_precision = True,
  main = "keras_utils_test.py",
  shard_count = 4,
  