diff --git a/src/finn/transformation/qonnx/fold_quant_weights.py b/src/finn/transformation/qonnx/fold_quant_weights.py index 12c854d3bab2b762abc3649e15beff29ff8de3ac..c81085e7a74eb3e08fbee447d7b9b027a436671e 100644 --- a/src/finn/transformation/qonnx/fold_quant_weights.py +++ b/src/finn/transformation/qonnx/fold_quant_weights.py @@ -103,7 +103,7 @@ class FoldQuantWeights(Transformation): model.set_initializer(node_out, q_node_output) else: # Check next operator type - mul_like_nodes = ["Mul", "Div", "Conv", "MatMul"] + mul_like_nodes = ["Mul", "Div", "Conv", "MatMul", "Gather"] add_like_nodes = ["Add", "Sub"] all_supported_ops = mul_like_nodes.copy() all_supported_ops.extend(add_like_nodes)