[−][src]Module tract_hir::internal
Re-exports
pub use super::prelude::*; |
pub use crate::infer::*; |
pub use crate::ops::binary::IntoHir; |
pub use crate::ops::expandable::expand; |
pub use crate::ops::expandable::Expansion; |
pub use shapefactoid; |
pub use to_typed; |
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. |
| 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. |