kime_tensor/lib.rs
1//! Device neutral tensor handles, dtypes, arenas, the `Backend` trait and the plan executor. A plan is a flat precomputed list of ops for one model at one shape bucket, and the hot path through it does no allocation, no locking and no dynamic dispatch per op. See spec/07-engine.md.
2//!
3//! One of the six crates where `unsafe` is allowed. Every block carries a `// SAFETY:` comment
4//! that names the invariant which makes it sound.
5
6pub mod backend;
7pub mod blob;
8pub mod bucket;
9pub mod dtype;
10pub mod plan;
11
12pub use backend::{Backend, Batch, BatchBuf, Caps, Error, Executor, HostTensor, Outputs, Result};
13pub use blob::Blob;
14pub use bucket::{Bucket, Buckets};
15pub use dtype::DType;
16pub use plan::{Epilogue, Graph, Layout, Op, Rows, Shape, Val, W, layout};