pytorch
c6cb99a6 - ns for fx: weight extraction for nni.ConvReLU2d (#54335)

Commit
3 years ago
ns for fx: weight extraction for nni.ConvReLU2d (#54335) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/54335 Simple fix to enable weight extraction for nni.ConvReLU2d. Note: this module only appears if the internal GraphModule APIs are called, so we add testing for this path. Test Plan: ``` python test/test_quantization.py TestFXNumericSuiteCoreAPIs.test_extract_weights_mod ``` Imported from OSS Reviewed By: hx89 Differential Revision: D27192844 fbshipit-source-id: 923cf63e29e4638fd77ca42e69aedb15fb20a330
Author
Parents
Loading