Skip to main content

luma_tensor/device/cpu/kernels/
mod.rs

1//! Generic CPU compute kernels, written over the [`element`] traits and plain
2//! slices + [`Layout`](crate::Layout). The `impl FloatOps/IntOps/BoolOps for Cpu`
3//! blocks dispatch the storage enums to these.
4
5pub mod element;
6pub mod elementwise;
7pub mod indexing;
8pub mod iter;
9pub mod matmul;
10pub mod nn;
11pub mod reduce;
12pub mod shape;