pytorch
5c431981 - OpInfo for `adaptive_avg_pool2d` (#62704)

Commit
3 years ago
OpInfo for `adaptive_avg_pool2d` (#62704) Summary: Please see https://github.com/facebookresearch/functorch/issues/78 and https://github.com/pytorch/pytorch/issues/54261. Note regarding sample inputs for this function: * Checks added for all relevant/interesting cases for `output_size`: `(None, None), (None, width), (height, None), (height, width)`. cc: mruberry zou3519 Chillee Pull Request resolved: https://github.com/pytorch/pytorch/pull/62704 Reviewed By: H-Huang Differential Revision: D30138788 Pulled By: zou3519 fbshipit-source-id: 66735ceaa85b9e6050d4ec27749fc3a8108cf557
Author
Parents
Loading