Source code
Revision control
Copy as Markdown
Other Tools
/* -*- Mode: IDL; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this file,
*/
[Func="InferenceSession::InInferenceProcess", Exposed=(DedicatedWorker,Window)]
interface Tensor {
[Throws]
constructor(UTF8String type,
(ArrayBufferView or sequence<any>) data,
sequence<long> dims);
[Cached, Pure]
attribute sequence<long> dims;
readonly attribute UTF8String type;
readonly attribute ArrayBufferView data;
readonly attribute TensorDataLocation location;
Promise<any> getData(optional boolean releaseData);
undefined dispose();
};
// Tensor Data Location
enum TensorDataLocation {
"none",
"cpu",
"cpu-pinned",
"texture",
"gpu-buffer",
"ml-tensor"
};
// Input/Output types
typedef record<UTF8String, Tensor> InferenceSessionTensorMapType;
typedef record<UTF8String, Tensor?> InferenceSessionNullableTensorMapType;
typedef InferenceSessionTensorMapType InferenceSessionFeedsType;
typedef (sequence<UTF8String> or InferenceSessionNullableTensorMapType) InferenceSessionFetchesType;
typedef InferenceSessionTensorMapType InferenceSessionReturnType;
dictionary InferenceSessionRunOptions {
unsigned short logSeverityLevel = 0; // 0 - 4
unsigned long logVerbosityLevel = 0;
boolean terminate = true;
UTF8String tag = "";
};
dictionary InferenceSessionSessionOptions {
sequence<any> executionProviders;
unsigned long intraOpNumThreads = 0;
unsigned long interOpNumThreads = 0;
record<UTF8String, unsigned long> freeDimensionOverrides;
UTF8String graphOptimizationLevel = "all";
boolean enableCpuMemArena = true;
boolean enableMemPattern = true;
UTF8String executionMode = "sequential";
UTF8String optimizedModelFilePath = "";
boolean enableProfiling = false;
UTF8String profileFilePrefix = "";
UTF8String logId = "";
unsigned short logSeverityLevel = 4; // 0 - 4
unsigned long logVerbosityLevel = 0;
(TensorDataLocation or record<UTF8String, TensorDataLocation>) preferredOutputLocation;
boolean enableGraphCapture = false;
record<UTF8String, any> extra;
};
[Func="InferenceSession::InInferenceProcess", Exposed=(DedicatedWorker,Window)]
interface InferenceSession {
[NewObject]
Promise<InferenceSessionReturnType> run(InferenceSessionFeedsType feeds, optional InferenceSessionRunOptions options = {});
[NewObject] static Promise<InferenceSession> create((UTF8String or Uint8Array) uriOrBuffer, optional InferenceSessionSessionOptions options = {});
[BinaryName=ReleaseSession]
Promise<undefined> release();
undefined startProfiling();
undefined endProfiling();
[Cached, Pure]
readonly attribute sequence<UTF8String> inputNames;
[Cached, Pure]
readonly attribute sequence<UTF8String> outputNames;
};