pytorch
2f0e4839 - [MPS] Fix correctness issues with Pooling ops (#91519)

Commit
1 year ago
[MPS] Fix correctness issues with Pooling ops (#91519) - Workaround for MaxPool when ceilMode=true - Workaround for ChannelsLast memory format - Workaround for divisor_override in AvgPool ops - Enabled count_include_pad parameter for AvgPool - Refactoring and clean up of duplicate code - Enable MaxPool tests in TestConsistency Pull Request resolved: https://github.com/pytorch/pytorch/pull/91519 Approved by: https://github.com/kulinseth, https://github.com/malfet
Author
Committer
Parents
Loading