ConversionContext

Context object for maintaining state during StableHLO conversion.

This class manages SSA value names, type mapping, and MLIR code generation during the conversion process from ComputeGraph to StableHLO.

Constructors

constructor(typeMapper: TypeMapper, graph: ComputeGraph? = null)

fun clear()

Clear all state (useful for testing)

Emit a comment with proper indentation

Emit a line of MLIR code with proper indentation

Emit an operation with proper indentation

Get the complete generated MLIR content

Get input nodes for a given node from the graph

Get the type mapper instance

Get the SSA value name for a node ID

Generate the next temporary SSA value name

Set the graph reference for node lookups

fun setValueName(nodeId: String, valueName: String)

Set the SSA value name for a node ID