fix pool bug, test=develop (#28359)

TCChenlong-patch-1
Double_V 5 years ago committed by GitHub
parent 6115c14fca
commit 17db031a05
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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

@ -503,6 +503,7 @@ class MaxPool2D(layers.Layer):
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 @@ class MaxPool3D(layers.Layer):
stride=self.stride,
padding=self.padding,
return_mask=self.return_mask,
ceil_mode=self.ceil_mode,
data_format=self.data_format,
name=self.name)

Loading…
Cancel
Save