pytorch
f6ae5db9 - [reland] OpInfo: `adaptive_avg_pool2d` (#62935)

Commit
3 years ago
[reland] OpInfo: `adaptive_avg_pool2d` (#62935) Summary: This PR is an attempt to reland https://github.com/pytorch/pytorch/pull/62704. **What has changed?** The op has non-deterministic behavior, hence an appropriate `gradcheck` wrapper had to be added. cc: mruberry zou3519 heitorschueroff kshitij12345 Pull Request resolved: https://github.com/pytorch/pytorch/pull/62935 Reviewed By: anjali411 Differential Revision: D30225095 Pulled By: zou3519 fbshipit-source-id: 644873cc21d44b19c8b68f9edff691913778de0e
Author
Committer
Parents
Loading