Module webgpu

Module webgpu 

Source
Expand description

WebGPU backend

Cross-platform GPU acceleration using wgpu. Works on both native (Vulkan/Metal/DX12) and WASM (WebGPU/WebGL).

Modules§

batchnorm
codegen
conv
dropout

Structs§

WebGpuBackend

Functions§

broadcast_to_webgpu
Broadcast operation on WebGPU (public API for dispatch)
elementwise_webgpu
Elementwise operations on WebGPU (public API for dispatch)
is_available_cached
Cached probe helper that performs a single adapter check and caches the result.
matmul_webgpu
Matrix multiplication on WebGPU (public API for dispatch)
reduction_webgpu
Reduction operations on WebGPU (public API for dispatch)