Update qbatch_norm2d opbenchmark test (#36630)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/36630
Test Plan:
OMP_NUM_THREADS=1 python -m pt.qbatchnorm_test
Imported from OSS
Differential Revision: D21030508
fbshipit-source-id: 1ece1bd7429207732eae4dd1982ceddcdc5d3a91
