mirror of https://github.com/llvm/torch-mlir
3ca35b4f3c
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. |
||
---|---|---|
.. | ||
main.py | ||
xfail_sets.py |