[nnc] Fix batchnorm implementation (#64112)
authorBert Maher <bertrand@fb.com>
Sun, 29 Aug 2021 02:18:10 +0000 (19:18 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Sun, 29 Aug 2021 02:20:35 +0000 (19:20 -0700)
commit4f969db325a7a70878bd3eae5bbb3fecd598d4ca
tree4026223c9fd30b643379c99dc71d15d632b77ba0
parentaefa2f3e643ab50fe4d8238ccdeb980143b6b454
[nnc] Fix batchnorm implementation (#64112)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/64112

Fixes #64062

Test Plan: Imported from OSS

Reviewed By: zhxchen17

Differential Revision: D30622897

Pulled By: bertmaher

fbshipit-source-id: 7d7c6131aa786e61fa1d0a517288396a0bdb1d22
test/test_jit_fuser_te.py
torch/csrc/jit/tensorexpr/operators/norm.cpp