fixing illegal memory access on NHWC BN kernel (#59981)
Summary:
adding an early exit in the kernel to avoid reading out of bound.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/59981
Reviewed By: ezyang
Differential Revision: D29147349
Pulled By: ngimel
fbshipit-source-id: b36a6a9e2526c609ff98fb5a44468f3257e0af67