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::TVec;
20    pub use crate::blob::Blob;
21    pub use crate::datum::{Datum, DatumType, QParams, round_ties_to_even};
22    pub use crate::dim::{Symbol, SymbolScope, SymbolValues, TDim, ToDim};
23    pub use crate::opaque::Opaque;
24    pub use crate::tensor::litteral::*;
25    pub use crate::tensor::{IntoArcTensor, IntoTensor, Tensor, natural_strides};
26    #[cfg(feature = "complex")]
27    pub use crate::tensor::{reinterpret_complex_as_inner_dim, reinterpret_inner_dim_as_complex};
28    pub use crate::tvec;
29    pub use crate::{TractError, TractResult};
30    pub use crate::{
31        dispatch_copy, dispatch_copy_by_size, dispatch_datum, dispatch_datum_by_size,
32        dispatch_floatlike, dispatch_hash, dispatch_numbers, dispatch_signed,
33    };
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::{DimLike, parse_tdim, solve_for};
43    pub use crate::opaque::{BlobWithFact, OpaqueFact, OpaquePayload};
44    pub use crate::prelude::*;
45    pub use crate::tensor::Approximation;
46    pub use crate::tensor::view::TensorView;
47    pub use crate::tensor::{clip_range_bounds, vector_size};
48    pub use anyhow::{Context as TractErrorContext, anyhow, bail, ensure, format_err};
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
