ONNX Gather not a graph node in ezkl? #236
Answered
by
alexander-camuto
biancaganescu
asked this question in
Q&A
-
I've been trying to use nn.Embedding with ezkl, but it does not work and I am not sure where to point the problem. Looking into it, ONNX maps nn.Embedding to the Gather operation (which I am not sure it gives the desired output anyways in that case) - from torch ONNX repo. Then ezkl does not recognise "Gather" as node 0 in the graph. sonos/tract say they implement "Gather", but I assume that is not a node for them so therefore that is not a node for ezkl? This is the ONNX graph created for this code, with input x = torch.ones(1, 64, dtype=int):
![]() |
Beta Was this translation helpful? Give feedback.
Answered by
alexander-camuto
May 16, 2023
Replies: 1 comment 1 reply
-
thanks for this let me take a look :) |
Beta Was this translation helpful? Give feedback.
1 reply
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
As discussed offline -- this has been resolved with latest updates.