pytorch
72c943a2 - ns for fx: fix bug for user function in weight extraction (#62333)

Commit
3 years ago
ns for fx: fix bug for user function in weight extraction (#62333) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/62333 We incorrectly ignored any custom relationships the user specified in the `extract_weights` API. Fixing this and adding a test case. Test Plan: ``` python test/test_quantization.py TestFXNumericSuiteCoreAPIs.test_user_defined_function ``` Imported from OSS Reviewed By: hx89 Differential Revision: D29963502 fbshipit-source-id: 33ce3d4df1acb6298b6c7dcb6674015c8d14bdf4
Author
Parents
Loading