pytorch
8dbf6ae8 - ns for fx: handling for user functions in weight and unshadowed act APIs (#56292)

Commit
3 years ago
ns for fx: handling for user functions in weight and unshadowed act APIs (#56292) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/56292 Adds hooks for specifying user defined functions to NS weight and unshadowed activation APIs. Adding it to shadowed activation APIs will be a bit more work, upcoming in a separate PR. Test Plan: ``` python test/test_quantization.py TestFXNumericSuiteCoreAPIs.test_user_defined_function ``` Imported from OSS Reviewed By: jerryzh168 Differential Revision: D27830409 fbshipit-source-id: 6bbddc3062c0b3e412a3147244795319c0785a92
Author
Parents
Loading