fix exampler and enable static test

Author: Asthestarsfalll

python/paddle/nn/layer/pooling.py

@@ -384,7 +384,7 @@ class LPPool2D(Layer):
 
  >>> # lp pool2d
  >>> input = paddle.uniform([1, 3, 32, 32], dtype="float32", min=-1, max=1)
- >>> AvgPool2D = nn.LPPool2D(norm_type=2, kernel_size=2, stride=2, padding=0)
+ >>> LPPool2D = nn.LPPool2D(norm_type=2, kernel_size=2, stride=2, padding=0)
  >>> output = LPPool2D(input)
  >>> print(output.shape)
  [1, 3, 16, 16]

test/legacy_test/test_pool2d_api.py

@@ -357,32 +357,6 @@ def check_avg_divisor(self, place):
  result = avg_pool2d_dg(input)
  np.testing.assert_allclose(result.numpy(), result_np, rtol=1e-05)
 
- def test_pool2d(self):
- for place in self.places:
- self.check_max_dygraph_results(place)
- self.check_avg_dygraph_results(place)
- self.check_max_dygraph_stride_is_none(place)
- self.check_avg_dygraph_stride_is_none(place)
- self.check_max_dygraph_padding(place)
- self.check_avg_divisor(place)
- self.check_max_dygraph_padding_results(place)
- self.check_max_dygraph_ceilmode_results(place)
- self.check_max_dygraph_nhwc_results(place)
- self.check_lp_dygraph_results(place)
- self.check_lp_dygraph_stride_is_none(place)
- self.check_lp_dygraph_ceilmode_results(place)
- self.check_lp_dygraph_nhwc_results(place)
- self.check_lp_dygraph_results_norm_type_is_inf(place)
- self.check_lp_dygraph_results_norm_type_is_negative_inf(place)
-
- @test_with_pir_api
- def test_pool2d_static(self):
- paddle.enable_static()
- for place in self.places:
- self.check_max_static_results(place)
- self.check_avg_static_results(place)
- paddle.disable_static()
-
  def check_lp_static_results(self, place):
  with paddle.static.program_guard(
  paddle.static.Program(), paddle.static.Program()
@@ -624,6 +598,33 @@ def check_lp_dygraph_stride_is_none(self, place):
  result = lp_pool2d_dg(input)
  np.testing.assert_allclose(result.numpy(), result_np, rtol=1e-05)
 
+ @test_with_pir_api
+ def test_pool2d_static(self):
+ paddle.enable_static()
+ for place in self.places:
+ self.check_max_static_results(place)
+ self.check_avg_static_results(place)
+ self.check_lp_static_results(place)
+ paddle.disable_static()
+
+ def test_pool2d(self):
+ for place in self.places:
+ self.check_max_dygraph_results(place)
+ self.check_avg_dygraph_results(place)
+ self.check_max_dygraph_stride_is_none(place)
+ self.check_avg_dygraph_stride_is_none(place)
+ self.check_max_dygraph_padding(place)
+ self.check_avg_divisor(place)
+ self.check_max_dygraph_padding_results(place)
+ self.check_max_dygraph_ceilmode_results(place)
+ self.check_max_dygraph_nhwc_results(place)
+ self.check_lp_dygraph_results(place)
+ self.check_lp_dygraph_stride_is_none(place)
+ self.check_lp_dygraph_ceilmode_results(place)
+ self.check_lp_dygraph_nhwc_results(place)
+ self.check_lp_dygraph_results_norm_type_is_inf(place)
+ self.check_lp_dygraph_results_norm_type_is_negative_inf(place)
+
 
 class TestPool2DError_API(unittest.TestCase):
  def test_error_api(self):