pytorch
b77c9791 - [quant][fx][graphmode] Make FixedQParam ops work for dtypes other than quint8 (#65484)

Commit
3 years ago
[quant][fx][graphmode] Make FixedQParam ops work for dtypes other than quint8 (#65484) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/65484 This PR makes sure we only use FixedQParamFakeQuantize for quint8 dtype and allows user to use other dtypes for ops like sigmoid, this is useful for producing reference pattern for these ops that can be used in other backends like TensorRT Test Plan: python test/test_quantization.py TestQuantizeFxOps Imported from OSS Reviewed By: vkuzo Differential Revision: D31120377 fbshipit-source-id: 3b529d588e2b6ff0377a89c181f6237f8f0cc2f5
Author
Parents
Loading