pytorch
98178757 - [quant][fx] Add support for BinarOpQuantizeHandler in backend_config_dict (#74882)

Commit
2 years ago
[quant][fx] Add support for BinarOpQuantizeHandler in backend_config_dict (#74882) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/74882 This PR adds support for ops like add/mul in backend_config_dict, these ops have different observation_type based on the number of tensor inputs, when number of tensor inputs is 1, we will share the output observer with input, otherwise we'll have a new observer. Test Plan: python test/test_quantization.py TestQuantizeFx python test/test_quantization.py TestQuantizeFxOps Imported from OSS Reviewed By: vkuzo, andrewor14 Differential Revision: D35236032 fbshipit-source-id: 7077f3ccee8a5d8d19b40107cf8ff16cceafc535 (cherry picked from commit a6f7a37f99fc727269d022d35cc5c0157b70c656)
Author
Committer
Parents
Loading