yscv-tensor 0.1.7

SIMD-accelerated tensor library with f32/f16/bf16 support and 80+ operations
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

//! Tensor and numeric primitives for the yscv framework.
#![allow(unsafe_code)]

pub const CRATE_ID: &str = "yscv-tensor";

#[path = "aligned.rs"]
mod aligned;
#[path = "error.rs"]
mod error;
#[path = "linalg.rs"]
mod linalg;
#[path = "ops.rs"]
mod ops;
#[path = "shape.rs"]
mod shape;
#[path = "simd/mod.rs"]
mod simd;
#[path = "tensor.rs"]
mod tensor;

pub use aligned::AlignedVec;
pub use error::{DType, TensorError};
pub use tensor::{Device, Tensor};

#[path = "tests/mod.rs"]
#[cfg(test)]
mod tests;