pytorch
92c7aec5 - ns for fx: add option to skip matching classes and functions (#56493)

Commit
3 years ago
ns for fx: add option to skip matching classes and functions (#56493) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/56493 Adds a config option to skip matching classes by class type and functions by function type. This is useful when users make custom modules which return types other than tensors. With the current implementation of Logger, these are not scriptable. Test Plan: ``` python test/test_quantization.py TestFXNumericSuiteCoreAPIs.test_user_module_scriptable ``` needs more testing before land Imported from OSS Reviewed By: jerryzh168 Differential Revision: D27886107 fbshipit-source-id: ec92c4f7ab7141021bc022f07b3b558b42bbb986
Author
Parents
Loading