torch.ScalarKernelEntry

export interface ScalarKernelEntry {
  readonly op: string;
  readonly dtype: DType;
  readonly device: Device;
  readonly cpu?: {
    readonly forward: ScalarCPUForwardFn;
  };
  readonly webgpu?: {
    readonly shader: string;
    readonly entryPoint: string;
    readonly workgroupSize?: readonly [number, number?, number?];
  };
  readonly predicate?: KernelPredicate;
}

readonlyop(string)
readonlydtype(DType)
readonlydevice(Device)
readonlycpu({ readonly forward: ScalarCPUForwardFn; })optional
readonlywebgpu({ readonly shader: string; readonly entryPoint: string; readonly workgroupSize?: readonly [number, number?, number?]; })optional
readonlypredicate(KernelPredicate)optional

Scalar kernel entry stored in registry.

torch.ScalarKernelEntry

export interface ScalarKernelEntry {
  readonly op: string;
  readonly dtype: DType;
  readonly device: Device;
  readonly cpu?: {
    readonly forward: ScalarCPUForwardFn;
  };
  readonly webgpu?: {
    readonly shader: string;
    readonly entryPoint: string;
    readonly workgroupSize?: readonly [number, number?, number?];
  };
  readonly predicate?: KernelPredicate;
}

readonlyop(string)
readonlydtype(DType)
readonlydevice(Device)
readonlycpu({ readonly forward: ScalarCPUForwardFn; })optional
readonlywebgpu({ readonly shader: string; readonly entryPoint: string; readonly workgroupSize?: readonly [number, number?, number?]; })optional
readonlypredicate(KernelPredicate)optional

Scalar kernel entry stored in registry.