From 5327712742ddef6d03c1fc6286e8302ca789336b Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Tue, 1 Feb 2022 09:48:56 +0100
Subject: [PATCH] [QONNX] support Quant -> Gather for quantized embeddings
---
src/finn/transformation/qonnx/fold_quant_weights.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/finn/transformation/qonnx/fold_quant_weights.py b/src/finn/transformation/qonnx/fold_quant_weights.py
index 12c854d3b..c81085e7a 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)
--
GitLab