[quant] Onnx export of quantized models with new API (#38736)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/38736
qconv2d and qlinear APIs were changed recently so updating the scale code accordingly
Test Plan:
python test/onnx/test_pytorch_onnx_caffe2_quantized.py
Imported from OSS
Differential Revision: D21647724
fbshipit-source-id: 45d4b358ffb84f1e73da8ba3f702d5043bdb16d2
