Interface MLTensorConstructorParameters<T>

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

Type Parameters

Hierarchy

  • CommonConstructorParameters<T>
  • GpuResourceConstructorParameters<T>
    • MLTensorConstructorParameters

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

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]>

Settings

Member Visibility

Theme

On This Page

Generated using TypeDoc