Crate burn_wgpu

Burn WGPU Backend

Modules

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

Macros

• binary_elemwise
  Creates a binary elementwise kernel.
• binary_elemwise_inplace
  Creates a binary elementwise inplace kernel.
• comparison
  Creates a comparison kernel.
• comparison_elem
  Creates a comparison elementwise kernel.
• comparison_elem_inplace
  Creates a comparison elementwise inplace kernel.
• comparison_inplace
  Creates a comparison inplace 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.
• unary_inplace
  Creates a unary inplace kernel.
• unary_scalar
  Creates a unary scalar kernel.
• unary_scalar_inplace
  Creates a unary scalar inplace kernel.

Structs

• AutoGraphicsApi
  Automatic graphics API based on OS.
• Dx11
  DirectX 11 graphics API.
• 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.