setValueType

fun setValueType(valueName: String, mlirType: String)(source)

Record the MLIR tensor type associated with an SSA value name.

Lets converters look up the declared type of an operand — the type it actually has when the op consumes it — instead of having to re-derive it from downstream node.inputs metadata, which can reflect a post-op shape rather than the operand's true shape. Seeded for %argN by StableHloConverter when the function signature is emitted, then populated for each op's result.