pytorch
808850b6 - [ARM] Do not use depthwise3x3 conv in grad mode (#56889)

Commit
3 years ago
[ARM] Do not use depthwise3x3 conv in grad mode (#56889) Summary: cpu_depthwise3x3_winograd is not grad aware and therefore should not be used if grad is expected on the input Fixes https://github.com/pytorch/pytorch/issues/56145 Pull Request resolved: https://github.com/pytorch/pytorch/pull/56889 Reviewed By: ngimel Differential Revision: D27990448 Pulled By: malfet fbshipit-source-id: 9c649f14b8f514eb1dfb7f0eb8e3357c09ddb299
Author
Parents
Loading