fix pool bug, test=develop (#28359)

PaddlePaddle · Nov 3, 2020 · 17db031 · 17db031
1 parent 6115c14
commit 17db031
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/python/paddle/nn/functional/pooling.py b/python/paddle/nn/functional/pooling.py
@@ -230,7 +230,7 @@ def avg_pool1d(x,
             x, 'pooling_type', 'avg', 'ksize', kernel_size, 'global_pooling',
             False, 'strides', stride, 'paddings', padding, 'padding_algorithm',
             padding_algorithm, 'use_cudnn', True, 'ceil_mode', ceil_mode,
-            'use_mkldnn', False, 'exclusive', not exclusive, 'data_format',
+            'use_mkldnn', False, 'exclusive', exclusive, 'data_format',
             data_format)
         return squeeze(output, [2])
 
@@ -338,7 +338,7 @@ def avg_pool2d(x,
             x, 'pooling_type', 'avg', 'ksize', kernel_size, 'global_pooling',
             False, 'padding_algorithm', padding_algorithm, 'strides', stride,
             'paddings', padding, 'use_cudnn', True, 'ceil_mode', ceil_mode,
-            'use_mkldnn', False, 'exclusive', not exclusive, 'data_format',
+            'use_mkldnn', False, 'exclusive', exclusive, 'data_format',
             data_format)
         if divisor_override is None:
             return output
@@ -452,7 +452,7 @@ def avg_pool3d(x,
             x, 'pooling_type', 'avg', 'ksize', kernel_size, 'strides', stride,
             'paddings', padding, 'global_pooling', False, 'padding_algorithm',
             padding_algorithm, 'use_cudnn', True, 'ceil_mode', ceil_mode,
-            'use_mkldnn', False, 'exclusive', not exclusive, 'data_format',
+            'use_mkldnn', False, 'exclusive', exclusive, 'data_format',
             data_format)
         if divisor_override is None:
             return output

diff --git a/python/paddle/nn/layer/pooling.py b/python/paddle/nn/layer/pooling.py
@@ -503,6 +503,7 @@ def forward(self, x):
             stride=self.stride,
             padding=self.padding,
             return_mask=self.return_mask,
+            ceil_mode=self.ceil_mode,
             data_format=self.data_format,
             name=self.name)
 
@@ -594,6 +595,7 @@ def forward(self, x):
             stride=self.stride,
             padding=self.padding,
             return_mask=self.return_mask,
+            ceil_mode=self.ceil_mode,
             data_format=self.data_format,
             name=self.name)