pytorch
8b992ab0 - ns for fx: add conv1d weight extraction (#55327)

Commit
3 years ago
ns for fx: add conv1d weight extraction (#55327) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/55327 Adds NS functionality for extracting weights from `F.conv1d` nodes. Test Plan: ``` python test/test_quantization.py TestFXNumericSuiteCoreAPIs.test_extract_weights_conv_fun ``` Imported from OSS Reviewed By: jerryzh168 Differential Revision: D27575425 fbshipit-source-id: 65fa194802ac7a9fb75b7616d962c5c2e71321ff
Author
Parents
Loading