diff --git a/tensorflow/compiler/tests/unary_ops_test.py b/tensorflow/compiler/tests/unary_ops_test.py index 8ffec3ad766..5b3f9465513 100644 --- a/tensorflow/compiler/tests/unary_ops_test.py +++ b/tensorflow/compiler/tests/unary_ops_test.py @@ -918,16 +918,12 @@ class UnaryOpsTest(xla_test.XLATestCase): np.array([1, 0x100000003f800000], np.int64), expected=np.array([1, 0x100000003f800000], np.uint64)) - @test_util.disable_mlir_bridge( - "TODO(b/153812660): Handle tf.InvertPermutation compilation") def testInvertPermutation(self): self._assertOpOutputMatchesExpected( array_ops.invert_permutation, np.array([1, 2, 0], np.int32), expected=np.array([2, 0, 1], dtype=np.int32)) - @test_util.disable_mlir_bridge( - "TODO(b/153812660): Handle tf.InvertPermutation compilation") def testInvertPermutationTwiceIsNoop(self): self._assertOpOutputMatchesExpected( lambda x: array_ops.invert_permutation(array_ops.invert_permutation(x)),