Module nn

Source

Modules§

memory
order
Evaluation order for nodes.
tract_data
tract_downcast_rs
Build status Latest version Documentation
tract_linalg
tract_ndarray
The ndarray crate provides an n-dimensional container for general elements and for numerics.
tract_num_traits
Numeric traits for generic mathematics
translator
typed

Macros§

anyhow
Construct an ad-hoc error from a string or existing non-anyhow error value.
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
dims
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
ensure
Return early with an error if a condition is not satisfied.
format_err
Construct an ad-hoc error from a string or existing non-anyhow error value.
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’.
AxesMapping
Axis
AxisChange
AxisChangeConsequence
BaseDataShape
Blob
DefaultRuntime
Graph
Main model class
HardSwish
HashMap
A hash map implemented with quadratic probing and SIMD lookup.
InletId
Identifier for a node input in the graph.
IntoShape
LeakyRelu
ModelPatch
A change to apply to a model.
Node
A Node in an Model.
Opaque
Outlet
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 T.
PlanOptions
Reduce
Rewriter
SessionState
ShapeFact
Sigmoid
SimplePlan
SimpleState
Softmax
Symbol
SymbolScope
SymbolValues
Tensor
Tensor is a concrete tensor in tract.
TensorView
TypedFact
Fully determined tensor information for TypedModel.
f16
A 16-bit floating point type implementing the IEEE 754-2008 standard binary16 a.k.a “half” format.

Enums§

Approximation
AxisOp
Cost
Cow
A clone-on-write smart pointer.
DataFormat
DatumType
GeometryBound
InOut
QParams
Reducer
SoftmaxExp
TDim
TValue
Validation
Level of precision to be expected in implementations comparisons.

Traits§

ClampCast
Datum
DatumExt
DatumTypeExt
DimLike
A super-trait for value acting as tensor dimensions in tract.
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.
FrozenOpState
Hash
A hashable type.
IntoArcTensor
Convenient conversion to Arc.
IntoTValue
IntoTensor
Convenient conversion to Tensor.
Op
A base operation
OpState
OpaqueFact
OpaquePayload
ResolveTo
Runnable
Runtime
SessionStateHandler
SpecialOps
State
ToDim
Convenience trait to convert values to TDim.
TractErrorContext
Provides the context method for Result.
TypedOp

Functions§

arr4
compute_shape_with_tf_rules
expand_mean_of_squares
hard_swish
leaky_relu
natural_strides
parse_tdim
perm_to_ops
rctensor0
rctensor1
rctensor2
rctensor3
rctensor4
round_ties_to_even
sigmoid
solve_for
tensor0
tensor1
tensor2
tensor3
tensor4
to_axis_ops_with_tf_rules
vector_size
wire_rank_broadcast
wire_with_rank_broadcast

Type Aliases§

DataShape
RunnableModel
A runnable model with fixed inputs and outputs.
SymDataShape
TVec
A Smallvec instantiation with 4 embeddable values.
TractError
TractResult
TypedFrozenSimpleState
An execution state for TypedModel, frozen (and Send).
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.

Derive Macros§

Hash
Derive macro generating an impl of the trait Hash.