pytorch
6249d730 - [ONNX] Fix export for avg_pool with default stride (#33017)

Commit
5 years ago
[ONNX] Fix export for avg_pool with default stride (#33017) Summary: If using nn.functional avg_pool, stride is an optional arg. If not provided, it is set to kernel_size. This PR fixes the export of avg_pool with default stride. Pull Request resolved: https://github.com/pytorch/pytorch/pull/33017 Reviewed By: hl475 Differential Revision: D19759604 Pulled By: houseroad fbshipit-source-id: b0352db6fbaf427f4cff9ba8a942efdeb39b6f02
Author
Parents
Loading