提交 5de6a191 编写于 作者: Z zhupengyang 提交者: Tao Luo

fix test_bilinear_tensor_product_op timeout (#22120)

上级 de56887b
...@@ -23,9 +23,9 @@ class TestBilinearTensorProductOp(OpTest): ...@@ -23,9 +23,9 @@ class TestBilinearTensorProductOp(OpTest):
def setUp(self): def setUp(self):
self.op_type = "bilinear_tensor_product" self.op_type = "bilinear_tensor_product"
batch_size = 6 batch_size = 6
size0 = 30 size0 = 5
size1 = 20 size1 = 4
size2 = 100 size2 = 5
a = np.random.random((batch_size, size0)).astype("float64") a = np.random.random((batch_size, size0)).astype("float64")
b = np.random.random((batch_size, size1)).astype("float64") b = np.random.random((batch_size, size1)).astype("float64")
w = np.random.random((size2, size0, size1)).astype("float64") w = np.random.random((size2, size0, size1)).astype("float64")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册