Pass in final value of reduce_range to quantize_initializer() function

microsoft · Nov 20, 2023 · c0b95f2 · c0b95f2
1 parent c452352
commit c0b95f2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/onnxruntime/python/tools/quantization/onnx_quantizer.py b/onnxruntime/python/tools/quantization/onnx_quantizer.py
@@ -924,7 +924,7 @@ def __quantize_inputs(
                         quant_type = quant_overrides["quant_type"].tensor_type
 
                     symmetric = quant_overrides.get("symmetric", self.is_weight_symmetric)
-                    reduce_range = quant_overrides.get("reduce_range", reduce_range)
+                    reduce_range = quant_overrides.get("reduce_range", reduce_range and self.reduce_range)
 
                     q_weight_name, zp_name, scale_name = self.quantize_initializer(
                         initializer, quant_type, reduce_range=reduce_range, symmetric=symmetric
@@ -1008,7 +1008,7 @@ def quantize_initializer(self, weight, qType, reduce_range=False, keep_float_wei
             w_data,
             qType,
             symmetric,
-            self.reduce_range and reduce_range,
+            reduce_range,
             self.min_real_range,
         )