From e413190cd65c6231656e8b571f806fd389dea27b Mon Sep 17 00:00:00 2001 From: danleifeng Date: Thu, 2 Sep 2021 09:25:57 +0000 Subject: [PATCH] fix flatten infershape; test=develop --- python/paddle/fluid/tests/unittests/test_flatten2_op.py | 2 +- .../fluid/tests/unittests/test_flatten_contiguous_range_op.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_flatten2_op.py b/python/paddle/fluid/tests/unittests/test_flatten2_op.py index e961e4da4b266..42b43cc46a69b 100644 --- a/python/paddle/fluid/tests/unittests/test_flatten2_op.py +++ b/python/paddle/fluid/tests/unittests/test_flatten2_op.py @@ -70,7 +70,7 @@ def init_test_case(self): self.new_shape = (36, 16) -class TestStaticFlattenPythonAPI(unittest.TestCase): +class TestStaticFlattenInferShapePythonAPI(unittest.TestCase): def execute_api(self, x, axis=1): return fluid.layers.flatten(x, axis=axis) diff --git a/python/paddle/fluid/tests/unittests/test_flatten_contiguous_range_op.py b/python/paddle/fluid/tests/unittests/test_flatten_contiguous_range_op.py index a335f584c7d4e..9093050d6d5c6 100644 --- a/python/paddle/fluid/tests/unittests/test_flatten_contiguous_range_op.py +++ b/python/paddle/fluid/tests/unittests/test_flatten_contiguous_range_op.py @@ -201,7 +201,7 @@ def test_static_api(self): self.assertTrue((2, 3, 16) == fetch_out[0].shape) -class TestStaticFlattenPythonAPI(unittest.TestCase): +class TestStaticFlattenInferShapePythonAPI(unittest.TestCase): def execute_api(self, x, start_axis=0, stop_axis=-1): return paddle.flatten(x, start_axis, stop_axis)