pytorch
00ca3635 - BatchNorm: Use resize_output and empty, instead of empty_like (#63084)

Commit
3 years ago
BatchNorm: Use resize_output and empty, instead of empty_like (#63084) Summary: Fixes https://github.com/pytorch/pytorch/issues/62967 This lets each of the three implementations choose which memory format to use for the output, meaning channels_last can be used in more cases. Pull Request resolved: https://github.com/pytorch/pytorch/pull/63084 Reviewed By: saketh-are Differential Revision: D30255740 Pulled By: ngimel fbshipit-source-id: 48d42850952ec910b29521a1c4e530eb2b29df5e
Author
Committer
Parents
Loading