skainet-io-onnx/sk.ainet.io.onnx

Package-level declarations

Types

data class OnnxAttributeValue(val name: String, val ints: List<Long> = emptyList(), val floats: List<Float> = emptyList(), val strings: List<String> = emptyList(), val tensor: TensorProto? = null)

OnnxBlobParseResult

class OnnxBlobParseResult(val metadata: ModelMetadata, val modelInfo: ModelInfo?, val tensors: List<TensorInfo>, val reader: StreamingOnnxReader?)

Browser-specific ONNX parsing result for WASM JS.

OnnxGraphView

data class OnnxGraphView(val nodes: List<OnnxNodeView>, val initializers: Map<String, OnnxTensorView>, val inputs: List<OnnxValueInfoView>, val outputs: List<OnnxValueInfoView>, val valueInfos: Map<String, OnnxValueInfoView>)

Minimal, proto-agnostic view of an ONNX graph. This keeps the importer independent of the generated proto surface while we wire in pbandk-generated classes.

OnnxLoadedModel

data class OnnxLoadedModel<M : Message>(val proto: M, val rawBytes: ByteArray)

Holder for the parsed ONNX model alongside its serialized form.

OnnxLoader

class OnnxLoader<M : Message>(readBytes: suspend () -> ByteArray, decode: (ByteArray) -> M)

Minimal pbandk-backed ONNX loader.

OnnxModelMetadata

data class OnnxModelMetadata(val irVersion: Long, val producerName: String?, val producerVersion: String?, val modelVersion: Long, val domain: String?, val docString: String?, val classNames: Map<Int, String>?, val numClasses: Int?, val description: String?, val author: String?, val license: String?, val task: String?, val rawMetadata: Map<String, String>)

Parsed model metadata from an ONNX file.

OnnxModelParser

class OnnxModelParser : BaseModelParser, AutoCloseable

ONNX model parser extending BaseModelParser. Provides metadata-first parsing of ONNX model files using SKaiNET's ONNX I/O capabilities.

OnnxModelReader

class OnnxModelReader : ModelReader

OnnxNodeView

data class OnnxNodeView(val name: String, val opType: String, val inputs: List<String>, val outputs: List<String>, val attributes: Map<String, OnnxAttributeValue>)

OnnxTensorView

data class OnnxTensorView(val name: String, val shape: List<Long>, val dtype: String, val proto: TensorProto)

OnnxToComputeGraphImporter

class OnnxToComputeGraphImporter

Stub importer that will map an ONNX graph view to a ComputeGraph. The mapping logic will be filled in as operator support lands.

OnnxValueInfoView

data class OnnxValueInfoView(val name: String, val shape: List<Int>?, val dtype: String?)

OnnxWeightLoader

object OnnxWeightLoader

Utility for loading ONNX model initializers (weights) into SKaiNET Module parameters.

StreamingOnnxReader

class StreamingOnnxReader : AutoCloseable

Streaming ONNX reader that parses metadata without loading tensor data.

StreamingOnnxTensorInfo

data class StreamingOnnxTensorInfo(val name: String, val dims: List<Long>, val dataType: Int, val dataTypeName: String, val nElements: Long, val rawDataOffset: Long, val rawDataLength: Int, val estimatedBytes: Int, val hasTypedArrayData: Boolean)

Tensor metadata for streaming access.