From 0a31b9ed3c2a3457264de1ef2f00e5afea02b4c4 Mon Sep 17 00:00:00 2001 From: Pranav Prakash Date: Thu, 17 Sep 2020 17:57:46 -0700 Subject: [PATCH] Fix order of returned values in quantize_weight_per_channel (#5205) Must match returned order of `quantize_inputs` --- onnxruntime/python/tools/quantization/onnx_quantizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxruntime/python/tools/quantization/onnx_quantizer.py b/onnxruntime/python/tools/quantization/onnx_quantizer.py index 8514058ff5..8b1d012d86 100644 --- a/onnxruntime/python/tools/quantization/onnx_quantizer.py +++ b/onnxruntime/python/tools/quantization/onnx_quantizer.py @@ -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: