pytorch
096adf4b - [quant][fx] Scope support for call_function in QuantizationTracer (#51086)

Commit
4 years ago
[quant][fx] Scope support for call_function in QuantizationTracer (#51086) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/51086 Previously we only supported getting scope for call_module and custom qconfig dict for call_module. This PR extends the Scope class to record the scope for all node types. For call_function qconfig if module_name is specified it takes precedence over function qconfig. Test Plan: python test/test_quantization.py test_qconfig_for_call_func Imported from OSS Reviewed By: jerryzh168 Differential Revision: D26077602 fbshipit-source-id: 99cdcdedde2280e51812db300e17d4e6d8f477d2
Author
Parents
Loading