Crate burn_wgpu

Structs

• AutoGraphicsApi
  Automatic graphics API based on OS.
• Dx12
  DirectX 12 graphics API.
• JitBackend
  Generic tensor backend that can be compiled just-in-time to any shader runtime
• JitTensor
  The basic tensor primitive struct.
• Metal
  Metal graphics API.
• OpenGl
  OpenGL graphics API.
• RuntimeOptions
  The values that control how a WGPU Runtime will perform its calculations.
• Vulkan
  Vulkan graphics API.
• WebGpu
  WebGpu graphics API.
• WgpuRuntime
  Runtime that uses the wgpu crate with the wgsl compiler.
• WorkGroup
  Provides launch information specifying the number of work groups to be used by a compute shader.

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.
• WgpuElement
  The base element trait for the wgpu backend.

Functions

• init_async
  Init the client async, necessary for wasm.
• init_sync
  Init the client sync, useful to configure the runtime options.
• select_device
  Select the wgpu device and queue based on the provided device.

Type Aliases

• Wgpu
  Tensor backend that uses the wgpu crate for executing GPU compute shaders.