saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-20 19:12:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix order of returned values in quantize_weight_per_channel (#5205)

Browse source 
Must match returned order of `quantize_inputs`
This commit is contained in:
Pranav Prakash 2020-09-17 17:57:46 -07:00 committed by Tianlei Wu
parent f726af34e0
commit 0a31b9ed3c
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/python/tools/quantization/onnx_quantizer.py
View file

@ -868,7 +868,7 @@ class ONNXQuantizer:
# Find if this input is already quantized
if weight_name in self.quantized_value_map:
quantized_value = self.quantized_value_map[weight_name]
return (quantized_value.zp_name, quantized_value.q_name, quantized_value.scale_name)
return (quantized_value.q_name, quantized_value.zp_name, quantized_value.scale_name)
else:
initializer = find_by_name(weight_name, self.model.initializer())
if initializer is None: