tract_data/
lib.rs

1#![allow(clippy::len_zero)]
2#![allow(clippy::missing_safety_doc)]
3
4pub extern crate itertools;
5
6#[macro_use]
7mod macros;
8
9/// A Smallvec instantiation with 4 embeddable values.
10///
11/// Used about everywhere in tract, for node inputs and outputs, or
12/// tensor dimensions.
13pub type TVec<T> = smallvec::SmallVec<[T; 4]>;
14
15pub type TractError = anyhow::Error;
16pub type TractResult<T> = anyhow::Result<T>;
17
18pub mod prelude {
19    pub use crate::blob::Blob;
20    pub use crate::datum::{round_ties_to_even, Datum, DatumType, QParams};
21    pub use crate::dim::{Symbol, SymbolScope, SymbolValues, TDim, ToDim};
22    pub use crate::opaque::Opaque;
23    pub use crate::tensor::litteral::*;
24    pub use crate::tensor::{natural_strides, IntoArcTensor, IntoTensor, Tensor};
25    #[cfg(feature = "complex")]
26    pub use crate::tensor::{reinterpret_complex_as_inner_dim, reinterpret_inner_dim_as_complex};
27    pub use crate::tvec;
28    pub use crate::TVec;
29    pub use crate::{
30        dispatch_copy, dispatch_copy_by_size, dispatch_datum, dispatch_datum_by_size,
31        dispatch_floatlike, dispatch_hash, dispatch_numbers, dispatch_signed,
32    };
33    pub use crate::{TractError, TractResult};
34    pub use half::f16;
35    pub use itertools as tract_itertools;
36    #[cfg(feature = "complex")]
37    pub use num_complex::Complex;
38}
39
40pub mod internal {
41    pub use crate::datum::ClampCast;
42    pub use crate::dim::{parse_tdim, solve_for, DimLike};
43    pub use crate::opaque::{OpaqueFact, OpaquePayload};
44    pub use crate::prelude::*;
45    pub use crate::tensor::vector_size;
46    pub use crate::tensor::view::TensorView;
47    pub use crate::tensor::Approximation;
48    pub use anyhow::{anyhow, bail, ensure, format_err, Context as TractErrorContext};
49    pub use ndarray as tract_ndarray;
50    pub use num_integer;
51    pub use num_traits as tract_num_traits;
52    pub use smallvec as tract_smallvec;
53    pub type StaticName = std::borrow::Cow<'static, str>;
54}
55
56pub use dim::TooEarly;
57pub use half;
58
59mod blob;
60mod datum;
61mod dim;
62mod opaque;
63mod scatter;
64mod tensor;
tract_data/lib.rs

tract_data/
lib.rs
