Class MetalKernel

Index

Constructors

Methods

Constructors

constructor

new MetalKernel(spec: MetalKernelSpec): MetalKernel
Parameters
- spec: MetalKernelSpec
Returns MetalKernel
- Defined in packages/mlx/src/metal-kernel.ts:53

Methods

apply

apply(inputs: MlxArray[], call: MetalKernelCall): MlxArray[]
Parameters
- inputs: MlxArray[]
- call: MetalKernelCall
Returns MlxArray[]
- Defined in packages/mlx/src/metal-kernel.ts:67

dispose

dispose(): void
Returns void
- Defined in packages/mlx/src/metal-kernel.ts:140