frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-11-13 21:59:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Make pool export compliant with onnx spec

Browse Source
This commit is contained in:
Dmytro Dzhulgakov
2017-12-04 09:40:44 -08:00
committed by Edward Z. Yang
parent 0d68ce9383
commit fcc142386b
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
torch/onnx/symbolic.py
View File

@ -276,7 +276,7 @@ def max_pool2d(g, input, kernel_size, stride, padding, dilation, ceil_mode):
stride = kernel_size
r = g.op("MaxPool", input,
kernel_shape_i=_pair(kernel_size),
pads_i=_pair(padding),
pads_i=_pair(padding) * 2,
strides_i=_pair(stride))
return r, None
@ -290,7 +290,7 @@ def avg_pool2d(g, input, kernel_size, stride, padding, ceil_mode, count_include_
return g.op("AveragePool", input,
kernel_shape_i=_pair(kernel_size),
strides_i=_pair(stride),
pads_i=_pair(padding))
pads_i=_pair(padding) * 2)
def avg_pool3d(g, input, kernel_size, stride, padding, ceil_mode, count_include_pad):