ONNX Gather not a graph node in ezkl?

[biancaganescu]

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):

class Model(nn.Module):
    def __init__(self):
        super().__init__()
        self.layer = nn. Embedding(65, 64)

    def forward(self, x):
        return self.layer(x)

And this is the ezkl error for it:

[alexander-camuto]

thanks for this let me take a look :)

[alexander-camuto]

As discussed offline -- this has been resolved with latest updates.