Module burn_wgpu::kernel

Kernel module

Modules

• conv
  Convolution kernels
• matmul
  Matmul kernels
• pool
  Pooling kernels
• prng
  Pseudo-random number generator kernels
• reduce
  Reduction algorithms

Macros

• binary
  Creates a binary kernel.
• kernel_wgsl
  Generates kernel source code by replacing some information using templating.
• unary
  Creates a unary kernel.

Structs

• CastRaw
  Generated kernel from wgsl file.
• ContiguousRaw
  Generated kernel from wgsl file.
• DynamicKernelSettings
  Generate kernel source code by replacing some information using templating.
• KernelSettings
  Generates kernel source code by replacing some information using templating.
• SourceTemplate
  Kernel source code abstraction allowing for templating.

Traits

• DynamicKernelSource
  Dynamic wgpu kernel to create a source template.
• StaticKernelSource
  Static wgpu kernel to create a source template.

Functions

• binary
  Launch an binary operation.
• build_info
  Create a vector containing the dimension, strides and shape of tensors.
• build_info_dyn
  Similar to build info but with dynamic rank.
• cast
  Cast a tensor to the given element type.
• into_contiguous
  Make a wgpu tensor contiguous.
• into_contiguous_dyn
  Similar to into contiguous but with dynamic rank.
• unary
  Launch an unary operation.