[−][src]Module tract_nnef::internal
Re-exports
pub use crate::ast::parse::parse_parameters; |
pub use crate::ast::FragmentDecl; |
pub use crate::ast::FragmentDef; |
pub use crate::ast::Parameter; |
pub use crate::ast::RValue; |
pub use crate::ast::param; |
pub use crate::ast::TypeName; |
pub use crate::deser::ModelBuilder; |
pub use crate::deser::ResolvedInvocation; |
pub use crate::framework::Nnef; |
pub use crate::registry::*; |
pub use crate::ser::invocation; |
pub use crate::ser::logical; |
pub use crate::ser::numeric; |
pub use crate::ser::IntoAst; |
Modules
order | Evaluation order for nodes. |
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 | Exits a function early with an error |
bin_to_super_type | |
dispatch_copy | |
dispatch_datum | |
dispatch_datum_by_size | |
dispatch_floatlike | |
dispatch_numbers | |
element_wise | |
element_wise_oop | |
not_a_pulsed_op | |
not_a_typed_op | |
op_as_typed_op | |
pulsed_op_to_typed_op | |
tvec |
Structs
Arc | A thread-safe reference-counting pointer. 'Arc' stands for 'Atomically Reference Counted'. |
AxisChange | |
AxisChangeConsequence | |
AxisInfo | Translation invariance property. |
AxisTracking | |
BaseNode | A Node in an Model. |
Blob | |
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. |
OutletFact | Information for each outlet of a node |
OutletId | Identifier for a node output in the graph. |
PhantomData | Zero-sized type used to mark things that "act like" they own a |
PulsedFact | |
SessionState | |
ShapeFact | Fully determined dimension of a tensor. |
SimplePlan | |
SimpleState | |
StreamFact | Streaming information for a streamed tensor. |
Tensor | Tensor is a concrete tensor in tract. |
TractError | The Error type. |
TypeId | A |
TypedFact | Fully determined tensor information for TypedModel. |
f16 |
Enums
AxisOp | |
Cost | |
Cow | A clone-on-write smart pointer. |
DatumType | |
InOut | |
TDim | |
TractErrorKind | The kind of an error. |
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 | |
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 | |
PulsedOp | |
SpecialOps | |
StatefullOp | |
StatelessOp | |
ToDim | Convenience trait to convert values to TDim. |
TractResultExt | Additional methods for |
TypedOp |
Functions
arr4 | |
change_axes | |
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 | |
perm_to_ops | |
rctensor0 | |
rctensor1 | |
rctensor2 | |
rctensor3 | |
rctensor4 | |
tensor0 | |
tensor1 | |
tensor2 | |
tensor3 | |
tensor4 |
Type Definitions
Node | |
PulsedModel | |
PulsedNode | |
RunnableModel | A runnable model with fixed inputs and outputs. |
TVec | A Smallvec instantiation with 4 embeddable values. |
TractResult | Convenient wrapper around |
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. |