pytorch
5973cbe6 - [Expanded Weights] fix conv3d (#80943)

Commit
3 years ago
[Expanded Weights] fix conv3d (#80943) Conv3d has an ordering typo in its unfold (it needs a custom unfold since unfold3d doesn't exist in torch). This was caught by Opacus but not by us because dilation, padding, and stride always matched in the test cases. Conv3d has very few test cases since it doesn't have an OpInfo and we were skipping some by the nn module testing filtering. So this updated this filtering to add more of the common nn tests (one of which did fail without the change) Closes #80953 Pull Request resolved: https://github.com/pytorch/pytorch/pull/80943 Approved by: https://github.com/zou3519
Author
samdow
Committer
Parents
Loading