Anonymous View

Interface MLTensorConstructorParameters<T>

interface MLTensorConstructorParameters<
    T extends MLTensorDataTypes = MLTensorDataTypes,
> {
    dims: readonly number[];
    location: "ml-tensor";
    mlTensor: MLTensorTypeFallback;
    type: T;
    dispose?(): void;
    download?(): Promise<DataTypeMap[T]>;
}

Type Parameters

T extends MLTensorDataTypes = MLTensorDataTypes

Hierarchy

CommonConstructorParameters<T>
GpuResourceConstructorParameters<T>
- MLTensorConstructorParameters

Index

Properties

dims location mlTensor type

Methods

dispose? download?

Properties

`Readonly`dims

dims: readonly number[]

Get the dimensions of the tensor.

`Readonly`location

location: "ml-tensor"

Specify the location of the data to be 'ml-tensor'.

`Readonly`mlTensor

mlTensor: MLTensorTypeFallback

Specify the WebNN MLTensor that holds the tensor data.

`Readonly`type

type: T

Specify the data type of the tensor.

Methods

`Optional`dispose

dispose?(): void
an optional callback function that will be called when the tensor is disposed.

If not provided, the tensor treat the GPU data as external resource.

Returns void
Inherited from GpuResourceConstructorParameters.dispose
- Defined in tensor-factory.ts:39

`Optional`download

download?(): Promise<DataTypeMap[T]>
an optional callback function to download data from GPU to CPU.

If not provided, the tensor treat the GPU data as external resource.

Returns Promise<DataTypeMap[T]>
Inherited from GpuResourceConstructorParameters.download
- Defined in tensor-factory.ts:32