torch-mlir/e2e_testing
Matthias Gehre 3ca35b4f3c
TorchToTosa: aten.embedding: Allow indices with any rank (#2327)
It's actually fine to not check the rank of the indices, because the conversion anyways flattens the index tensor to be (1, numElements) before applying tosa::gather, and then anyways reshapes the output tensor to the output shape of the aten.embedding.
2023-07-21 08:54:19 +02:00
..
main.py Add make_fx_tosa variant to end2end tests (#2240) 2023-07-13 15:07:54 +02:00
xfail_sets.py TorchToTosa: aten.embedding: Allow indices with any rank (#2327) 2023-07-21 08:54:19 +02:00