pytorch
430fc03e - ns for fx: add category for ops which accept fp32 or int8 input (#55859)

Commit
3 years ago
ns for fx: add category for ops which accept fp32 or int8 input (#55859) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/55859 Adds mappings for ops which can accept either fp32 or int8 input, such as `F.relu`. A future PR will fill out the op coverage. Test Plan: ``` python test/test_quantization.py TestFXNumericSuiteCoreAPIs.test_op_with_either_fp32_or_int8_input ``` Imported from OSS Reviewed By: raghuramank100 Differential Revision: D27740659 fbshipit-source-id: cfc3dd58319b7161ca7f1fe05cd22d9a3ff11141
Author
Parents
Loading