Preserve memory layout when aten batchnorm is used (#62773)
Summary:
https://github.com/pytorch/pytorch/issues/62594
CC cpuhrsch
Pull Request resolved: https://github.com/pytorch/pytorch/pull/62773
Reviewed By: H-Huang
Differential Revision: D30118658
Pulled By: cpuhrsch
fbshipit-source-id: bce9e92f5f8710c876a33cccbd1625155496ddea
