Crate burn_wgpu

Burn WGPU Backend

Modules

• compute
  Compute related module.
• kernel
  Kernel module
• tensor
  Tensor module.

Macros

• binary
  Creates a binary kernel.
• kernel_wgsl
  Generates kernel source code by replacing some information using templating.
• reduce_tune_ops
  Generate an autotune operation for a reduce kernel
• unary
  Creates a unary kernel.

Structs

• AutoGraphicsApi
  Automatic graphics API based on OS.
• Dx12
  DirectX 12 graphics API.
• Metal
  Metal graphics API.
• OpenGl
  OpenGL graphics API.
• Vulkan
  Vulkan graphics API.
• WebGpu
  WebGpu graphics API.
• Wgpu
  Tensor backend that uses the wgpu crate for executing GPU compute shaders.

Enums

• WgpuDevice
  The device struct when using the wgpu backend.

Traits

• FloatElement
  The float element type for the wgpu backend.
• GraphicsApi
  The basic trait to specify which graphics API to use as Backend.
• IntElement
  The int element type for the wgpu backend.