pytorch
64b4e37c - ns for fx: allow graph matching of parents of cat (#52368)

Commit
3 years ago
ns for fx: allow graph matching of parents of cat (#52368) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/52368 Before this PR, the graph matching logic only handles node arguments of type Node. This PR extends it to allow to handle node arguments of type Tuple, so that the matcher can properly navigate through the arguments of `cat`. Test Plan: ``` python test/test_quantization.py TestFXGraphMatcher.test_nodes_before_cat ``` Imported from OSS Reviewed By: raghuramank100 Differential Revision: D26490101 fbshipit-source-id: 2de8d6acc30f237e22bfc3cfa89728b37411aab6
Author
Parents
Loading