Module burn_jit::kernel

Kernel module

Modules

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

Macros

• binary
  Creates a binary kernel.
• unary
  Creates a unary kernel.

Enums

• MaskFillStrategy
  Define how to run the mask fill kernel.
• MaskWhereStrategy
  Define how to run the mask where kernel.

Traits

• GpuComputeShaderPhase
  Dynamic jit kernel to create a compute shader.

Functions

• binary
  Launch an binary operation.
• bool_cast
  Cast a bool tensor to the given element type.
• cast
  Cast a tensor to the given element type.
• into_contiguous
  Make a jit tensor contiguous.
• mask_fill
  Execute the mask fill kernel with the given strategy.
• mask_where
  Execute the mask where kernel with the given strategy.
• unary
  Launch an unary operation.