pytorch
7b42ecad - [Pytorch] Disable use of qnnpack with ceil_mode avgpool (#79028)

Commit
2 years ago
[Pytorch] Disable use of qnnpack with ceil_mode avgpool (#79028) Summary: In pooling ops qnnpack does not handle ceil_mode. This is protected in maxpool but not in avgpool. This diff adds that. Test Plan: CI Differential Revision: D36975661 Pull Request resolved: https://github.com/pytorch/pytorch/pull/79028 Approved by: https://github.com/dzdang
Author
Committer
Parents
Loading