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)