Skip to content
Snippets Groups Projects
Commit 53277127 authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[QONNX] support Quant -> Gather for quantized embeddings

parent 2d5d5fd2
No related branches found
No related tags found
No related merge requests found
......@@ -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)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment