pytorch
80b1b7e4 - ns for fx: ensure kwargs are handled when graph matching (#55078)

Commit
4 years ago
ns for fx: ensure kwargs are handled when graph matching (#55078) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/55078 Fixes a TODO, make sure we iterate through kwargs as well as args when navigating graphs. We can use `node.all_input_nodes` convenience property to accomplish this. Test Plan: ``` python test/test_quantization.py TestFXGraphMatcher ``` Imported from OSS Reviewed By: jerryzh168 Differential Revision: D27474699 fbshipit-source-id: 8a6e3db5a73328c4f296ac5fce951e81213b6f58
Author
Parents
Loading