pytorch
1618dc2a - ns for fx: update graph matching to handle dicts and tuples in node args (#52681)

Commit
3 years ago
ns for fx: update graph matching to handle dicts and tuples in node args (#52681) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/52681 Updates the NS graph matching to properly traverse through args of nodes if args are lists or tuples. As a side benefit, refactors the code to make future similar improvements easier. Test Plan: ``` python test/test_quantization.py TestFXGraphMatcher ``` Imported from OSS Reviewed By: raghuramank100 Differential Revision: D26611221 fbshipit-source-id: 4ddd9b26338a5a2763b2883967e100f73e207538
Author
Parents
Loading