未验证 提交 163a3ddf 编写于 作者: D Double_V 提交者: GitHub

fix row_conv_op small size unittest cases (#22606)

上级 e1366613
......@@ -50,7 +50,7 @@ class TestRowConvOp1(OpTest):
lod = [[2, 3, 2]]
T = sum(lod[0])
D = 16
context_length = 2
context_length = 8
x = np.random.random((T, D)).astype("float32")
wt = np.random.random((context_length, D)).astype("float32")
......@@ -139,11 +139,11 @@ def row_conv_foward_Tensor(x, wt):
class TestRowOpWithTensorInput(OpTest):
def setUp(self):
self.op_type = "row_conv"
length = [3, 2, 4]
length = [1, 2, 3]
B = 2
T = sum(length)
D = 16
context_length = 2
D = 20
context_length = 6
x = np.random.random((B, T, D)).astype("float32")
wt = np.random.random((context_length, D)).astype("float32")
......
......@@ -25,7 +25,6 @@ NEED_TO_FIX_OP_LIST = [
'margin_rank_loss',
'matmul',
'mul',
'row_conv',
'scatter',
'smooth_l1_loss',
'soft_relu',
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册