StableHloConverter

class StableHloConverter(registry: StableHloOperationRegistry, typeMapper: TypeMapper, validator: MlirValidator? = null)(source)

Main converter class that orchestrates the conversion process from ComputeGraph to StableHLO MLIR.

This class provides a modular architecture for converting computational graphs to StableHLO format, using a registry-based system for operation mapping and a conversion context for state management.

Constructors

Link copied to clipboard
constructor(registry: StableHloOperationRegistry, typeMapper: TypeMapper, validator: MlirValidator? = null)

Functions

Link copied to clipboard
fun convert(graph: ComputeGraph, functionName: String = "main"): StableHloModule

Convert a ComputeGraph to StableHLO MLIR format

Link copied to clipboard

Convert with optimization passes applied