pytorch
2d7c1e3f - [bc-breaking] Produce quantization pattern for add_scalar and mul_scalar (#61859)

Commit
3 years ago
[bc-breaking] Produce quantization pattern for add_scalar and mul_scalar (#61859) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/61859 BC-breakign note: Previously we do not add observer/fake_quant for output of add/mul for tensor - scalar operation, in this PR we added the observer/fake_quant instance (that's the same as input) to correctly model the behavior of the quantized add_scalar and mul_scalar op (since quantized add/mul scalar assumes the output quantized tensor have the same quantization parameter as input) Test Plan: python test/test_quantization.py TestQuantizeFxOps.test_add python test/test_quantization.py TestQuantizeFxOps.test_mul Imported from OSS Reviewed By: vkuzo Differential Revision: D29770859 fbshipit-source-id: f43fcbfecd04c392467770b22c481bbbdaf43c25
Author
Parents
Loading