diff --git a/tensorflow/python/tpu/BUILD b/tensorflow/python/tpu/BUILD index 01f0e95c0b9..9cd75d1bed7 100644 --- a/tensorflow/python/tpu/BUILD +++ b/tensorflow/python/tpu/BUILD @@ -551,7 +551,6 @@ tpu_py_test( "tpu_outside_compilation_test.py", ], disable_experimental = True, - disable_mlir_bridge = False, python_version = "PY3", tags = ["no_oss"], deps = [ diff --git a/tensorflow/python/tpu/tpu_outside_compilation_test.py b/tensorflow/python/tpu/tpu_outside_compilation_test.py index d09a2f70088..5ae618640b2 100644 --- a/tensorflow/python/tpu/tpu_outside_compilation_test.py +++ b/tensorflow/python/tpu/tpu_outside_compilation_test.py @@ -32,7 +32,6 @@ from tensorflow.python.eager import remote from tensorflow.python.eager import test from tensorflow.python.framework import config from tensorflow.python.framework import constant_op -from tensorflow.python.framework import test_util from tensorflow.python.lib.io import tf_record from tensorflow.python.ops import array_ops from tensorflow.python.ops import control_flow_ops @@ -555,9 +554,6 @@ class OutsideCompilationOnUnsupportedOpTest(test.TestCase, self.assertEqual(events[1].summary.value[0].tag, "cond/x") self.assertEqual(events[1].summary.value[0].simple_value, 3.0) - @test_util.disable_mlir_bridge( - "TODO(b/168493455): Reenable this test one deadlock resolved." - ) def testAutoOutsideCompilationWithFunctionalNodes(self): strategy = get_tpu_strategy()