Expand description
Kernel module
Modules§
- conv
- Convolution kernels
- interpolate
- Interpolation kernels
- matmul
- Matmul kernels
- pool
- Pooling kernels
- prng
- Pseudo-random number generator kernels
- quantization
- Quantization operations
- reduce
- Reduction algorithms
Enums§
- Mask
Fill Strategy - Define how to run the mask fill kernel.
- Mask
Where Strategy - Define how to run the mask where kernel.
Constants§
- PLANE_
DIM_ APPROX - An approximation of the plane dimension.
Traits§
- Kernel
- Implement this trait to create a kernel definition.
Functions§
- 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.