pytorch
55b10e6b - [Pytorch][Vulkan] Use specalized shader for 3x3 depthwise conv (#89953)

Commit
2 years ago
[Pytorch][Vulkan] Use specalized shader for 3x3 depthwise conv (#89953) This diff uses specialized implementation for 3x3 and 5x5 dw conv. Differential Revision: [D41006638](https://our.internmc.facebook.com/intern/diff/D41006638/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/89953 Approved by: https://github.com/salilsdesai, https://github.com/kirklandsign
Author
Committer
Parents
Loading