pytorch
e0b82d7d - [MPS] Fix convolution `Source and weight input channels mismatch' crash (#91822)

Commit
1 year ago
[MPS] Fix convolution `Source and weight input channels mismatch' crash (#91822) Fixes crashes in conv input/weight backward passes due to NCHW / NHWC formats. Pull Request resolved: https://github.com/pytorch/pytorch/pull/91822 Approved by: https://github.com/razarmehr
Author
Committer
Parents
Loading