pytorch
a7dc0ab8 - [quant][fx][pyper] Get first linear use of quantize_per_tensor for FQN (#54859)

Commit
4 years ago
[quant][fx][pyper] Get first linear use of quantize_per_tensor for FQN (#54859) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/54859 This is applicable to the case when a call_function linear op is one of the users of quantize op In order to be able to map the qparams of quantize_per_tensor to the qparams of the linear operator that consumes it, we need to use the FQN of the module with linear op for the qparmas of quantize_per_tensor. Test Plan: python test/test_quantization.py test_qparams_fqn Imported from OSS Reviewed By: jerryzh168 Differential Revision: D27390505 fbshipit-source-id: a47af0e5ac016f2b2df74fbdf45afe99dc04be46
Author
Parents
Loading