Skip to content

Commit 27346a7

Browse files
Update Gather op without input_type
Internal IR remove "input_type" from Gather op. This commit removes "input_type" setting and set Gather inputs by Co-authored-by: Jingya Zhang <jingya.zhang@samsung.com> Signed-off-by: jiseong.oh <jiseong.oh@samsung.com>
1 parent aef65c1 commit 27346a7

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

backends/samsung/builders/op_embedding.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,8 +35,8 @@ def define_node(
3535

3636
output_id = self.define_tensor(node, enn_graph, vals_to_ids)
3737

38-
params = {"axis": 0, "input_type": "indices"}
38+
params = {"axis": 0}
3939
self._update_params_qdtype(node, params)
4040
enn_graph.define_op(
41-
node.name, "GATHER", [input_id, weight_id], [output_id], params
41+
node.name, "GATHER", [weight_id, input_id], [output_id], params
4242
)

0 commit comments

Comments
 (0)