import { Env, InferenceSession, Tensor } from 'onnxruntime-common'; /** * tuple elements are: ORT element type; dims; tensor data */ export declare type SerializableTensor = [Tensor.Type, readonly number[], Tensor.DataType]; /** * tuple elements are: InferenceSession handle; input names; output names */ export declare type SerializableSessionMetadata = [number, string[], string[]]; interface MessageError { err?: string; } interface MessageInitWasm extends MessageError { type: 'init-wasm'; in?: Env.WebAssemblyFlags; } interface MessageInitOrt extends MessageError { type: 'init-ort'; in?: { numThreads: number; loggingLevel: number; }; } interface MessageCreateSession extends MessageError { type: 'create'; in?: { model: Uint8Array; options?: InferenceSession.SessionOptions; }; out?: SerializableSessionMetadata; } interface MessageReleaseSession extends MessageError { type: 'release'; in?: number; } interface MessageRun extends MessageError { type: 'run'; in?: { sessionId: number; inputIndices: number[]; inputs: SerializableTensor[]; outputIndices: number[]; options: InferenceSession.RunOptions; }; out?: SerializableTensor[]; } interface MesssageEndProfiling extends MessageError { type: 'end-profiling'; in?: number; } export declare type OrtWasmMessage = MessageInitWasm | MessageInitOrt | MessageCreateSession | MessageReleaseSession | MessageRun | MesssageEndProfiling; export {};