fuse layernorm + quantize (#44232)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/44232
enhance layernorm to optionally quantize its output
add fusion code to replace instances of layernorm +quantization
Test Plan:
tested layernorm
net_runner
P141557987
Reviewed By: venkatacrc
Differential Revision: D23510893
fbshipit-source-id: 32f57ba2090d35d86dcc951e0f3f6a8901ab3153
