From 8aa3df0f972e1d046d5a87c2a75bc4276b3a0882 Mon Sep 17 00:00:00 2001 From: Smit Hinsu Date: Fri, 12 Jun 2020 19:21:21 -0700 Subject: [PATCH] Enable InvertPermutation test with MLIR bridge Lowering pattern for InvertPermutation is fixed now. PiperOrigin-RevId: 316218452 Change-Id: I80f45227f8c77ab8fba2066579d4640216f9968b --- tensorflow/compiler/tests/unary_ops_test.py | 4 ---- 1 file changed, 4 deletions(-) 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)),