stensor 0.3.1

Cross-platform GPU tensor library with Slang and Rust.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

//! Fundamental linear-algebra matrix/vector operations.

mod contiguous;
mod gemm;
mod gemv;
mod op_assign;
mod reduce;
mod repeat;

pub use contiguous::Contiguous;
pub use gemm::{Gemm, GemmVariant};
pub use gemv::{Gemv, GemvVariant, MatrixMode, N, T};
pub use op_assign::{BinOpOffsets, OpAssign, OpAssignVariant};
pub use reduce::{Reduce, ReduceVariant};
pub use repeat::Repeat;