Crate burn::backend::wgpu

Structs

• AutoGraphicsApi
  Automatic graphics API based on OS.
• CubeDim
• 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. This is used in the Wgpu backend. For advanced configuration, use init_sync to pass in runtime options or to select a specific graphics API.

Enums

• WgpuDevice
  The device struct when using the wgpu backend.

Traits

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

Functions

• init_async
  Like init_sync, but async, necessary for wasm.
• init_existing_device
• init_sync
  Initialize a client on the given device with the given options. This function is useful to configure the runtime options or to pick a different graphics API. On wasm, it is necessary to use init_async instead.
• 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.