update flatten class api kernal name test=develop

PaddlePaddle · Aug 4, 2020 · ee096b1 · ee096b1
1 parent bd770a1
commit ee096b1
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 94 deletions.
diff --git a/python/paddle/fluid/dygraph/nn.py b/python/paddle/fluid/dygraph/nn.py
@@ -3228,7 +3228,7 @@ def forward(self, input):
         out = self._helper.create_variable_for_type_inference(input.dtype)
         x_shape = self._helper.create_variable_for_type_inference(input.dtype)
         self._helper.append_op(
-            type="flatten2",
+            type="flatten_contiguous_range",
             inputs={"X": input},
             outputs={"Out": out,
                      "XShape": x_shape},

diff --git a/python/paddle/fluid/tests/unittests/test_flatten_2d_op.py b/python/paddle/fluid/tests/unittests/test_flatten_2d_op.py