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