sony / mct_quantizers

Apache License 2.0
17 stars 8 forks source link

Fix issue in per-tensor quantization and missing channel_axis #64

Closed reuvenperetz closed 1 year ago

reuvenperetz commented 1 year ago

When None is passed as channel_axis, the op has no attribute of channel_axis, which creates conflict with the onnxruntime function. For this reason, if we quantize per-tensor and channel_axis is None, we set it to 0.