mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/61377 Both the quantization tracer and the NS tracer record `_node_name_to_scope`, which contains the mapping from node name to FQN. This PR adds the FQN information to the NS results, so that it is more convenient for users to attribute a NS result to the corresponding module in their model. Test Plan: ``` python test/test_quantization.py TestFXNumericSuiteCoreAPIs.test_extract_weights_fqn python test/test_quantization.py TestFXNumericSuiteCoreAPIs.test_match_activations_fqn python test/test_quantization.py TestFXNumericSuiteCoreAPIs.test_shadow_activations_fqn ``` Imported from OSS Reviewed By: jerryzh168 Differential Revision: D29600349 fbshipit-source-id: df489e03daff97dd380f59c83ffdc2b0012a0a53 |
||
|---|---|---|
| .. | ||
| fx | ||
| ns | ||
| __init__.py | ||
| _correct_bias.py | ||
| _equalize.py | ||
| _learnable_fake_quantize.py | ||
| _numeric_suite_fx.py | ||
| _numeric_suite.py | ||
| fake_quantize.py | ||
| fuse_modules.py | ||
| fuser_method_mappings.py | ||
| observer.py | ||
| qconfig.py | ||
| quant_type.py | ||
| quantization_mappings.py | ||
| quantize_fx.py | ||
| quantize_jit.py | ||
| quantize.py | ||
| stubs.py | ||
| utils.py | ||