pytorch
d73e36a4 - ns for fx: change API to take nn.Module instead of GraphModule (#53075)

Commit

3 years ago

ns for fx: change API to take nn.Module instead of GraphModule (#53075) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/53075 The input and output types should be `nn.Module`, to hide the implementation detail that the pass is using FX. Test Plan: ``` python test/test_quantization.py TestFXNumericSuiteCoreAPIs ``` Imported from OSS Reviewed By: jerryzh168 Differential Revision: D26740548 fbshipit-source-id: d5ed445379355bebdd90d377c95fcd7e671371a3

Author

vkuzo

Committer

facebook-github-bot

Parents

b00cdfe1

Files1

torch/quantization/ns
- numeric_suite_core_apis_fx.py

pytorch d73e36a4 - ns for fx: change API to take nn.Module instead of GraphModule (#53075)

pytorch
d73e36a4 - ns for fx: change API to take nn.Module instead of GraphModule (#53075)