More LayerNorm Vectorization in calcMeanStd function. (#41618)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/41618
More LayerNorm Vectorization in calcMeanStd function.
Test Plan: test covered in test_layernorm_nnpi_fp16.py
Reviewed By: hyuen
Differential Revision: D22606585
fbshipit-source-id: be773e62f0fc479dbc2d6735f60c2e98441916e9
