pytorch
534c8215 - fix bn channels_last contiguity check (#50659)

Commit
3 years ago
fix bn channels_last contiguity check (#50659) Summary: Fixes https://github.com/pytorch/pytorch/issues/42588 The contiguity check used to be for memory format suggested by `grad_output->suggest_memory_format()`, but an invariant guaranteed by derivatives.yaml is `input->suggest_memory_format()` Pull Request resolved: https://github.com/pytorch/pytorch/pull/50659 Reviewed By: mruberry Differential Revision: D25938921 Pulled By: ngimel fbshipit-source-id: a945bfef6ce3d91b17e7ff96babe89ffd508939a
Author
Natalia Gimelshein
Parents
Loading