diff --git a/python/paddle/fluid/tests/unittests/test_row_conv_op.py b/python/paddle/fluid/tests/unittests/test_row_conv_op.py index 602f650a5027726ae788399a70688bd0a9d48055..59caa25405fc49ef274918e0ab7f19f4c33444ab 100644 --- a/python/paddle/fluid/tests/unittests/test_row_conv_op.py +++ b/python/paddle/fluid/tests/unittests/test_row_conv_op.py @@ -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") diff --git a/python/paddle/fluid/tests/unittests/white_list/check_shape_white_list.py b/python/paddle/fluid/tests/unittests/white_list/check_shape_white_list.py index 7cc5445e02d6c5238b06964dcf5c4c00ba7ac881..9061daa0fe7958dc3057f40daf8b637baf371595 100644 --- a/python/paddle/fluid/tests/unittests/white_list/check_shape_white_list.py +++ b/python/paddle/fluid/tests/unittests/white_list/check_shape_white_list.py @@ -25,7 +25,6 @@ NEED_TO_FIX_OP_LIST = [ 'margin_rank_loss', 'matmul', 'mul', - 'row_conv', 'scatter', 'smooth_l1_loss', 'soft_relu',