diff --git a/python/paddle/fluid/tests/unittests/mkldnn/test_reshape_mkldnn_op.py b/python/paddle/fluid/tests/unittests/mkldnn/test_reshape_mkldnn_op.py index da1b338de62b6440cbf9a802335a50b207095244..ecb9b70216cee427907b4d6962842eabbf566270 100644 --- a/python/paddle/fluid/tests/unittests/mkldnn/test_reshape_mkldnn_op.py +++ b/python/paddle/fluid/tests/unittests/mkldnn/test_reshape_mkldnn_op.py @@ -28,7 +28,7 @@ from paddle.fluid.tests.unittests.test_reshape_op import TestReshapeOp paddle.enable_static() -class TestReshapeOneDNNOp(TestReshapeOp): +class TestReshape2OneDNNOp(TestReshapeOp): def setUp(self): super().setUp() self.attrs['use_mkldnn'] = True