Add LazyBatchNormXd (#51862)
Summary:
Same diff with https://github.com/pytorch/pytorch/issues/51548 (cc. albanD)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/51862
Reviewed By: izdeby
Differential Revision: D26312289
Pulled By: albanD
fbshipit-source-id: 9cdec0e0c9021c33d10d85010978c7fa5cb4dc60
