pytorch
2a2bc1fc - ns for fx: add fqn to results, when present (#61377)

Commit
4 years ago
ns for fx: add fqn to results, when present (#61377) 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
Author
Parents
Loading