skainet-compile-hlo/sk.ainet.compile.hlo/StableHloModule

StableHloModule

data class StableHloModule(val content: String, val functionName: String = "main", val inputSpecs: List<TensorSpec> = emptyList(), val outputSpecs: List<TensorSpec> = emptyList(), val metadata: Map<String, Any> = emptyMap(), val externalParameters: List<ExternalParameterRef> = emptyList())(source)

Represents a StableHLO MLIR module text output

Constructors

StableHloModule

constructor(content: String, functionName: String = "main", inputSpecs: List<TensorSpec> = emptyList(), outputSpecs: List<TensorSpec> = emptyList(), metadata: Map<String, Any> = emptyMap(), externalParameters: List<ExternalParameterRef> = emptyList())

Properties

content

val content: String

externalParameters

val externalParameters: List<ExternalParameterRef>

Constants that were lifted out of content as util.global references under ConstantMaterializationPolicy.ExternalAlways. A downstream packager writes these into an IREE .irpa; see issue #523. Empty under the default inline policy.

functionName

val functionName: String

inputSpecs

val inputSpecs: List<TensorSpec>

metadata

val metadata: Map<String, Any>

outputSpecs

val outputSpecs: List<TensorSpec>