[quant] Support 2 dim input in quantized batchnorm 1d (#51597)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/51597
aliging quantized batchnorm behavior with fp batchnorm
Test Plan:
python test/test_quantization.py TestQuantizedOps.test_batch_norm
python test/test_quantization.py TestQuantizedOps.test_batch_norm_relu
Imported from OSS
Reviewed By: vkuzo
Differential Revision: D26212489
fbshipit-source-id: 663d5d70cc82ea5cc68e66452590efe1342998f1
