pytorch
cada2cd3 - [ONNX] Support per channel quantization

Commit
2 years ago
[ONNX] Support per channel quantization Extending the support for quantization with per channel quantization. An extra attribute `axis` can be found for per channel quantized tensors, most commonly in quantized weight of Convolution or Linear module. The PR adds support to correctly parse the `axis` attribute, and map to ONNX representation in `QuantizeLinear` and `DequantizeLinear`. Pull Request resolved: https://github.com/pytorch/pytorch/pull/76002 Approved by: https://github.com/garymm
Author
Committer
Parents
Loading