pytorch
b524a110 - ns for fx: add ref_node_target_type (#62685)

Commit
3 years ago
ns for fx: add ref_node_target_type (#62685) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/62685 Adds a `ref_node_target_type` field to hold the string type of the base node. This is needed because in some cases the previous node does not match ref_node (if we have observers, or if we are logging inputs), and it is useful to know the type of ref_node. Test Plan: ``` python test/test_quantization.py TestFXNumericSuiteCoreAPIs ``` Imported from OSS Reviewed By: hx89 Differential Revision: D30082947 fbshipit-source-id: 98ded7b25a5d8d5ea820e0ef62c3799b65c3fc77
Author
Parents
Loading