burn_cubecl/kernel/
mod.rs

1mod binary;
2mod binary_int;
3mod cast;
4mod clamp;
5mod comparison;
6mod contiguous;
7mod index;
8mod mask;
9mod unary_float;
10mod unary_int;
11mod unary_numeric;
12
13pub(crate) use binary::*;
14pub(crate) use binary_int::*;
15pub use cast::*;
16pub use contiguous::*;
17pub use mask::*;
18pub(crate) use unary_float::*;
19pub(crate) use unary_int::*;
20pub(crate) use unary_numeric::*;
21
22pub use burn_common::PLANE_DIM_APPROX;
23pub use cubecl::Kernel;
24
25/// Convolution kernels
26pub mod conv;
27/// Interpolation kernels
28pub mod interpolate;
29/// Matmul kernels
30pub mod matmul;
31/// Pooling kernels
32pub mod pool;
33/// Pseudo-random number generator kernels
34pub mod prng;
35/// Quantization operations
36pub mod quantization;
37/// Reduction algorithms
38pub mod reduce;
39
40pub(crate) use clamp::*;
41pub(crate) use comparison::*;
42pub use index::*;
burn_cubecl/kernel/mod.rs

burn_cubecl/kernel/
mod.rs
