burn_jit/kernel/
mod.rs

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

burn_jit/kernel/
mod.rs
