pytorch
444b318a - ns for fx: add linear-relu mod weight extraction (#55080)

Commit
3 years ago
ns for fx: add linear-relu mod weight extraction (#55080) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/55080 Adds support for extracting weights of linear-relu module pattern. Test Plan: ``` python test/test_quantization.py TestFXNumericSuiteCoreAPIs ``` Imported from OSS Reviewed By: raghuramank100 Differential Revision: D27474701 fbshipit-source-id: 69ceaadc28d7fdcebd16d519367274d348b0dd29
Author
Parents
Loading