skainet-compile-hlo/sk.ainet.compile.hlo.converters/ShapeOperationsConverter

ShapeOperationsConverter

class ShapeOperationsConverter : StableHloOperationConverter(source)

Converter for shape manipulation operations.

This converter implements shape operations using StableHLO primitives:

reshape: using stablehlo.reshape with proper shape inference
flatten: using stablehlo.reshape to flatten specified dimensions
squeeze: using stablehlo.reshape to remove singleton dimensions
unsqueeze: using stablehlo.broadcast_in_dim for dimension expansion

Supports operations as specified in Requirements 2.5:

Shape operations (reshape, flatten, squeeze, unsqueeze)
Dynamic reshaping with runtime shape computation
Proper shape inference and validation

Constructors

ShapeOperationsConverter

constructor()

Properties

supportedOperations

open override val supportedOperations: Set<String>

Set of operation names this converter supports

Functions

open override fun convert(node: GraphNode, operands: List<String>, context: ConversionContext): ConversionResult

Convert a graph node to StableHLO operations