TE is a limited declarative programming model, it’s not possible to write operations that do data-dependent indexing in TE.
Anything that’s sort/scatter related needs to be written directly in the more imperative TIR.
TE is a limited declarative programming model, it’s not possible to write operations that do data-dependent indexing in TE.
Anything that’s sort/scatter related needs to be written directly in the more imperative TIR.