[TFRT] Add support for transposed inputs in tf.MatMul native op/kernel
PiperOrigin-RevId: 334257561 Change-Id: I8ab8351bc4a4c814a1aadca73a0638163d2a5171
This commit is contained in:
parent
534cb9ab79
commit
15484c9758
@ -77,8 +77,6 @@ class OpsTest(test_util.TensorFlowTestCase, parameterized.TestCase):
|
|||||||
total = math_ops.add_n([three, four])
|
total = math_ops.add_n([three, four])
|
||||||
self.assertAllEqual(7, total)
|
self.assertAllEqual(7, total)
|
||||||
|
|
||||||
@test_util.disable_tfrt('b/163564975: TFRT MatMul Kernel is partially '
|
|
||||||
'implemented.')
|
|
||||||
def testExecuteBoolAttr(self):
|
def testExecuteBoolAttr(self):
|
||||||
three = constant_op.constant([[3]])
|
three = constant_op.constant([[3]])
|
||||||
five = constant_op.constant([[5]])
|
five = constant_op.constant([[5]])
|
||||||
|
Loading…
x
Reference in New Issue
Block a user