mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-28 03:20:58 +00:00
1ab8a95eb6
* support session options and run options. use onnxruntime c api. * fix lint errors * add an error code on throwing an exception * resolve comments. change remaining C++ APIs to C API
59 lines
2.6 KiB
TypeScript
59 lines
2.6 KiB
TypeScript
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
export interface OrtWasmModule extends EmscriptenModule {
|
|
//#region emscripten functions
|
|
stackSave(): number;
|
|
stackRestore(stack: number): void;
|
|
stackAlloc(size: number): number;
|
|
|
|
UTF8ToString(offset: number): string;
|
|
lengthBytesUTF8(str: string): number;
|
|
stringToUTF8(str: string, offset: number, maxBytes: number): void;
|
|
//#endregion
|
|
|
|
//#region ORT APIs
|
|
_OrtInit(numThreads: number, loggingLevel: number): number;
|
|
|
|
_OrtCreateSession(dataOffset: number, dataLength: number, sessionOptionsHandle: number): number;
|
|
_OrtReleaseSession(sessionHandle: number): void;
|
|
_OrtGetInputCount(sessionHandle: number): number;
|
|
_OrtGetOutputCount(sessionHandle: number): number;
|
|
_OrtGetInputName(sessionHandle: number, index: number): number;
|
|
_OrtGetOutputName(sessionHandle: number, index: number): number;
|
|
|
|
_OrtFree(stringHandle: number): void;
|
|
|
|
_OrtCreateTensor(dataType: number, dataOffset: number, dataLength: number, dimsOffset: number, dimsLength: number):
|
|
number;
|
|
_OrtGetTensorData(tensorHandle: number, dataType: number, dataOffset: number, dimsOffset: number, dimsLength: number):
|
|
number;
|
|
_OrtReleaseTensor(tensorHandle: number): void;
|
|
_OrtRun(
|
|
sessionHandle: number, inputNamesOffset: number, inputsOffset: number, inputCount: number,
|
|
outputNamesOffset: number, outputCount: number, outputsOffset: number, runOptionsHandle: number): number;
|
|
|
|
_OrtCreateSessionOptions(): number;
|
|
_OrtReleaseSessionOptions(sessionOptionsHandle: number): void;
|
|
_OrtSetSessionGraphOptimizationLevel(sessionOptionsHandle: number, level: number): number;
|
|
_OrtEnableCpuMemArena(sessionOptionsHandle: number): number;
|
|
_OrtDisableCpuMemArena(sessionOptionsHandle: number): number;
|
|
_OrtEnableMemPattern(sessionOptionsHandle: number): number;
|
|
_OrtDisableMemPattern(sessionOptionsHandle: number): number;
|
|
_OrtSetSessionExecutionMode(sessionOptionsHandle: number, mode: number): number;
|
|
_OrtSetSessionLogId(sessionOptionsHandle: number, logid: number): number;
|
|
_OrtSetSessionLogSeverityLevel(sessionOptionsHandle: number, level: number): number;
|
|
|
|
_OrtCreateRunOptions(): number;
|
|
_OrtReleaseRunOptions(runOptionsHandle: number): void;
|
|
_OrtRunOptionsSetRunLogSeverityLevel(runOptionsHandle: number, level: number): number;
|
|
_OrtRunOptionsSetRunTag(runOptionsHandle: number, tag: number): number;
|
|
//#endregion
|
|
|
|
//#region config
|
|
mainScriptUrlOrBlob?: string|Blob;
|
|
//#endregion
|
|
}
|
|
|
|
declare const moduleFactory: EmscriptenModuleFactory<OrtWasmModule>;
|
|
export default moduleFactory;
|