torch.DeviceRegistry
export interface DeviceRegistry {
cpu: 'cpu';
webgpu: 'webgpu';
}cpu('cpu')webgpu('webgpu')
Core type definitions for the torch.library extensibility system.
This module provides the foundational types for operation schemas, enabling type-safe registration of kernels, autograd, and custom dtypes.