pytorch
7a69e3d3 - [fx][subgraph_matcher] Add a matcher that supports name to node map (#110743)

Commit
1 year ago
[fx][subgraph_matcher] Add a matcher that supports name to node map (#110743) Summary: We want the matcher to return a name -> node in target graph so that we can refer to the node by name, this is useful for downstream applications like quantization. and also we can use the torch API as source of truth instead of matching aten API directly. Test Plan: python test/fx/test_matcher_utils.py Reviewers: Subscribers: Tasks: Tags: Pull Request resolved: https://github.com/pytorch/pytorch/pull/110743 Approved by: https://github.com/SherlockNoMad
Author
Committer
Parents
Loading