ReactantExt: Einsum -> DotGeneral

[wsmoses]

Tenet.jl/ext/TenetReactantExt.jl

Line 170 in 2bba50c

einsum_config = Reactant.MLIR.IR.Attribute("$(join(tia)),$(join(tib))->$(join(tic))")

So:

einsum

This operation is on its way out of StableHLO, so it is not included in the StableHLO specification: https://github.com/openxla/stablehlo/issues/3.

Informally, this operation does the same thing as TF's einsum: https://www.tensorflow.org/api_docs/python/tf/einsum

Presumably this can be rewritten as dot general and the likes?

[mofeing]

yes, I'm aware of that. the problem is that einsum is more general than dot_general as it can also involve transpose, some diagonal extraction and a sum over some dimensions.

I need to build a parser for it and call the ops accordingly, but it's a burden. do you know where that string parsing is implemented in StableHLO or XLA?