Optimize LayerNormOp (#17604)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17604
Optimize LayerNormOp
i-am-not-moving-c2-to-c10
Reviewed By: houseroad
Differential Revision: D14274175
fbshipit-source-id: a7aa263a1b0eb109682d2be99306e7b2cdcc0faf