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 crate::runtime::Runtime;
pub use crate::runtime::Runnable;
pub use crate::runtime::State;
pub use crate::runtime::DefaultRuntime;
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
ndarray
crate 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-
anyhow
error value. - Return early with an error.
- Return early with an error if a condition is not satisfied.
- Construct an ad-hoc error from a string or existing non-
anyhow
error 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
binary16
a.k.a “half” format.
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
context
method forResult
.
Functions§
Type Aliases§
- A Smallvec instantiation with 4 embeddable values.
Derive Macros§
- Derive macro generating an impl of the trait
Hash
.