pytorch
13d7b40e - ns for fx: add F.conv2d and F.conv3d weight extraction (#55287)

Commit
4 years ago
ns for fx: add F.conv2d and F.conv3d weight extraction (#55287) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/55287 Adds support for extracting weights from F.conv2d and F.conv3d. F.conv1d and the fused variants are saved for future PRs. Test Plan: ``` python test/test_quantization.py TestFXNumericSuiteCoreAPIs.test_extract_weights_conv_fun ``` Imported from OSS Reviewed By: hx89 Differential Revision: D27575424 fbshipit-source-id: e945912d7d0ab320f47cab30d00d60ddb7497158
Author
Parents
Loading