1 2 3 4 5 6 7 8 9 10 11
//! CPU kernels — one sub-module per family. pub mod attention; pub mod conv2d; pub mod elementwise; pub mod layernorm; pub mod matmul; pub mod quant; pub mod reduce; pub mod rope; pub mod softmax;