Module tract_core::internal
source · Expand description
This prelude is meant for code extending tract (like implementing new ops).
Re-exports
pub use crate::axes::AxesMapping;pub use crate::axes::Axis;pub use crate::ops::element_wise::ElementWiseMiniOp;pub use crate::ops::Cost;pub use crate::ops::EvalOp;pub use crate::ops::FrozenOpState;pub use crate::ops::Op;pub use crate::ops::OpState;pub use crate::ops::Validation;pub use crate::plan::SessionState;pub use downcast_rs as tract_downcast_rs;pub use crate::model::*;pub use crate::ops::change_axes::*;pub use crate::prelude::*;
Modules
- Integer trait and functions.
- Extra iterator adaptors, functions and macros.
- The
ndarraycrate provides an n-dimensional container for general elements and for numerics. - Numeric traits for generic mathematics
- Small vectors in various sizes. These store a certain number of elements inline, and fall back to the heap for larger allocations. This can be a useful optimization for improving cache locality and reducing allocator traffic for workloads that fit within the inline buffer.
Macros
- Construct an ad-hoc error from a string or existing non-
anyhowerror value. - Return early with an error.
- Construct an ad-hoc error from a string or existing non-
anyhowerror value.
Structs
- A hash map implemented with quadratic probing and SIMD lookup.
- Zero-sized type used to mark things that “act like” they own a
T. - Tensor is a concrete tensor in tract.
- A 16-bit floating point type implementing the IEEE 754-2008 standard
binary16a.k.ahalfformat.
Enums
- A clone-on-write smart pointer.
Traits
- A super-trait for value acting as tensor dimensions in tract.
- A hashable type.
- Convenient conversion to Arc
. - Convenient conversion to Tensor.
- Convenience trait to convert values to TDim.
- Provides the
contextmethod forResult.
Functions
Type Definitions
- A Smallvec instantiation with 4 embeddable values.
Derive Macros
- Derive macro generating an impl of the trait
Hash.