pytorch
0fbc2be2 - ns for fx: enable `call_method` nodes in graph matching (#56194)

Commit
3 years ago
ns for fx: enable `call_method` nodes in graph matching (#56194) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/56194 Enables the NS graph matcher to also match `call_method` nodes. These are useful for ops such as `torch.sigmoid`. Test Plan: ``` python test/test_quantization.py TestFXGraphMatcher.test_methods ``` Imported from OSS Reviewed By: jerryzh168 Differential Revision: D27805333 fbshipit-source-id: 509ae283db6b245671f11e3eb6b7fcb3a5735ef5
Author
Parents
Loading