We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6d5e970 commit d419df4Copy full SHA for d419df4
onnxruntime/python/tools/quantization/onnx_quantizer.py
@@ -942,7 +942,7 @@ def _dequantize_value(self, value_name):
942
self.model.model.producer_name == "onnx-quantizer" and scale_init is not None
943
):
944
# axis is not specified so scale_init must be a scalar.
945
- assert onnx.numpy_helper.to_array(scale_init).size == 1
+ assert scale_init is None or onnx.numpy_helper.to_array(scale_init).size == 1
946
947
dqlinear_name = value_name + "_DequantizeLinear"
948
dqlinear_node = self.model.find_node_by_name(dqlinear_name, self.new_nodes, self.model.graph())
0 commit comments