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

Commit
4 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
Parents
Loading