未验证 提交 22e6b9e3 编写于 作者: S ShenLiang 提交者: GitHub

Fix the ut of matmulv2 for broadcast case (#29461)

* fix the ut of matmulv2 for broadcast
上级 a5fcc4b5
......@@ -244,8 +244,8 @@ class TestMatMuklOp14(TestMatMulV2Op):
"""
def config(self):
self.x_shape = (3, 1, 1, 10, 10)
self.y_shape = (1, 2, 2, 10, 10)
self.x_shape = (3, 1, 6, 6)
self.y_shape = (1, 2, 6, 9)
self.trans_x = True
self.trans_y = False
......@@ -256,8 +256,8 @@ class TestMatMuklOp15(TestMatMulV2Op):
"""
def config(self):
self.x_shape = (3, 1, 1, 10, 10)
self.y_shape = (1, 2, 2, 10, 10)
self.x_shape = (3, 1, 6, 6)
self.y_shape = (1, 2, 6, 9)
self.trans_x = False
self.trans_y = False
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册