From d398b12c32e551b07a845f4bdf00ac16b926fb9a Mon Sep 17 00:00:00 2001 From: narutozxp <41829090+narutozxp@users.noreply.github.com> Date: Thu, 28 May 2026 20:55:46 +0800 Subject: [PATCH] fix(qonnx): handle scalar zpt for per-tensor bias quantization --- src/qonnx/transformation/extract_conv_bias.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/qonnx/transformation/extract_conv_bias.py b/src/qonnx/transformation/extract_conv_bias.py index 14973d9d..a357d34d 100644 --- a/src/qonnx/transformation/extract_conv_bias.py +++ b/src/qonnx/transformation/extract_conv_bias.py @@ -78,7 +78,7 @@ def apply(self, model): if quant_scale.shape != (1,): model.set_initializer(producer.input[1], quant_scale.reshape(add_shape)) quant_zpt = model.get_initializer(producer.input[2]) - if quant_zpt.shape != (1,): + if quant_zpt.size != 1: model.set_initializer(producer.input[2], quant_zpt.reshape(add_shape)) model.set_tensor_shape(producer.output[0], add_shape)