Module tract_nnef::internal [−][src]
Re-exports
pub use crate::ast::parse::parse_parameters; |
pub use crate::ast::param; |
pub use crate::ast::FragmentDecl; |
pub use crate::ast::FragmentDef; |
pub use crate::ast::Parameter; |
pub use crate::ast::RValue; |
pub use crate::ast::TypeName; |
pub use crate::deser::ModelBuilder; |
pub use crate::deser::ResolvedInvocation; |
pub use crate::framework::Nnef; |
pub use crate::prelude::*; |
pub use crate::registry::*; |
pub use crate::ser::invocation; |
pub use crate::ser::logical; |
pub use crate::ser::numeric; |
pub use crate::ser::string; |
pub use crate::ser::IntoAst; |
Modules
order | Evaluation order for nodes. |
tract_data | |
tract_downcast_rs | |
tract_itertools | Extra iterator adaptors, functions and macros. |
tract_linalg | |
tract_ndarray | The |
tract_num_traits | Numeric traits for generic mathematics |
translator | |
typed |
Macros
Hash | Derive macro generating an impl of the trait |
args_1 | |
args_2 | |
args_3 | |
args_4 | |
args_5 | |
args_6 | |
args_7 | |
args_8 | |
as_op | |
bail | Return early with an error. |
bin_to_super_type | |
dispatch_copy | |
dispatch_copy_by_size | |
dispatch_datum | |
dispatch_datum_by_size | |
dispatch_floatlike | |
dispatch_hash | |
dispatch_numbers | |
dispatch_signed | |
element_wise | |
element_wise_oop | |
format_err | Construct an ad-hoc error from a string or existing non- |
impl_dyn_hash | |
impl_op_same_as | |
not_a_typed_op | |
op_as_typed_op | |
tvec |
Structs
Arc | A thread-safe reference-counting pointer. ‘Arc’ stands for ‘Atomically Reference Counted’. |
AxisChange | |
AxisChangeConsequence | |
AxisInfo | Translation invariance property. |
AxisTracking | |
Blob | |
Dims | |
Graph | Main model class |
HashMap | A hash map implemented with quadratic probing and SIMD lookup. |
InletId | Identifier for a node input in the graph. |
Invariants | |
ModelPatch | A change to apply to a model. |
Node | A Node in an Model. |
Outlet | Information for each outlet of a node |
OutletId | Identifier for a node output in the graph. |
OutletMap | |
PhantomData | Zero-sized type used to mark things that “act like” they own a |
SessionState | |
ShapeFact | Fully determined dimension of a tensor. |
SimplePlan | |
SimpleState | |
Symbol | |
SymbolValues | |
Tensor | Tensor is a concrete tensor in tract. |
TensorView | |
TypeId | A |
TypedFact | Fully determined tensor information for TypedModel. |
f16 |
Enums
AttrOrInput | |
AxisOp | |
Cost | |
Cow | A clone-on-write smart pointer. |
DatumType | |
InOut | |
TDim | |
Validation | Level of precision to be expected in implementations comparisons. |
Traits
Datum | |
DimLike | A super-trait for value acting as tensor dimensions in tract. |
DynHash | |
ElementWiseMiniOp | |
EvalOp | |
Fact | Type information about a tensor: shape, and element type, in various state of determination. |
Framework | A Framework that translate its model to tract core model. |
Hash | A hashable type. |
IntoArcTensor | Convenient conversion to Arc |
IntoTensor | Convenient conversion to Tensor. |
MaybeProduct | |
Op | A base operation |
OpState | |
SloppyHash | |
SpecialOps | |
ToDim | Convenience trait to convert values to TDim. |
TractErrorContext | Provides the |
TypedOp |
Functions
arr4 | |
change_axes | |
dyn_hash | |
eval_order | Find an evaluation order for a model, using its default inputs and outputs as boundaries. |
for_model | |
full_axis_tracking | |
hash_f32 | |
hash_opt_f32 | |
natural_strides | |
perm_to_ops | |
rctensor0 | |
rctensor1 | |
rctensor2 | |
rctensor3 | |
rctensor4 | |
tensor0 | |
tensor1 | |
tensor2 | |
tensor3 | |
tensor4 |
Type Definitions
RunnableModel | A runnable model with fixed inputs and outputs. |
TVec | A Smallvec instantiation with 4 embeddable values. |
TractError | |
TractResult | |
TypedModel | A model with completely determined types and shapes. |
TypedModelPatch | A ModelPatch for TypedModel. |
TypedNode | Node for TypedModel graph |
TypedRunnableModel | A runnable TypedModel (new name for SimplePlan). |
TypedSimplePlan | An execution plan for TypedModel. |
TypedSimpleState | An execution state for TypedModel. |