From 22e6b9e373d6e602287cd146351c6781c513db4a Mon Sep 17 00:00:00 2001 From: ShenLiang Date: Tue, 8 Dec 2020 14:33:18 +0800 Subject: [PATCH] Fix the ut of matmulv2 for broadcast case (#29461) * fix the ut of matmulv2 for broadcast --- python/paddle/fluid/tests/unittests/test_matmul_v2_op.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_matmul_v2_op.py b/python/paddle/fluid/tests/unittests/test_matmul_v2_op.py index 75d82d27024..1695058f7b3 100644 --- a/python/paddle/fluid/tests/unittests/test_matmul_v2_op.py +++ b/python/paddle/fluid/tests/unittests/test_matmul_v2_op.py @@ -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 -- GitLab