experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Enable InvertPermutation test with MLIR bridge

Browse Source 
Lowering pattern for InvertPermutation is fixed now.

PiperOrigin-RevId: 316218452
Change-Id: I80f45227f8c77ab8fba2066579d4640216f9968b
This commit is contained in:
Smit Hinsu 2020-06-12 19:21:21 -07:00 committed by TensorFlower Gardener
parent 36140531e8
commit 8aa3df0f97
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/compiler/tests/unary_ops_test.py
View File

@ -918,16 +918,12 @@ class UnaryOpsTest(xla_test.XLATestCase):
np.array([1, 0x100000003f800000], np.int64), np.array([1, 0x100000003f800000], np.int64),
expected=np.array([1, 0x100000003f800000], np.uint64)) expected=np.array([1, 0x100000003f800000], np.uint64))
@test_util.disable_mlir_bridge(
"TODO(b/153812660): Handle tf.InvertPermutation compilation")
def testInvertPermutation(self): def testInvertPermutation(self):
self._assertOpOutputMatchesExpected( self._assertOpOutputMatchesExpected(
array_ops.invert_permutation, array_ops.invert_permutation,
np.array([1, 2, 0], np.int32), np.array([1, 2, 0], np.int32),
expected=np.array([2, 0, 1], dtype=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): def testInvertPermutationTwiceIsNoop(self):
self._assertOpOutputMatchesExpected( self._assertOpOutputMatchesExpected(
lambda x: array_ops.invert_permutation(array_ops.invert_permutation(x)), lambda x: array_ops.invert_permutation(array_ops.invert_permutation(x)),