Struct tract_core::ops::source::Source

pub struct Source;

Methods

impl Source

pub fn new() -> Self

Constructs a new Source.

Trait Implementations

impl InferenceRulesOp for Source

fn rules<'r, 'p: 'r, 's: 'r>(
    &'s self,
    _s: &mut Solver<'r>,
    inputs: &'p [TensorProxy],
    outputs: &'p [TensorProxy]
) -> InferenceResult

Registers the inference rules of the operator.

fn as_op(&self) -> &dyn Op

fn as_op_mut(&mut self) -> &mut dyn Op

fn to_typed(
    &self,
    _source: &InferenceModel,
    node: &InferenceNode,
    target: &mut TypedModel,
    _mapping: &HashMap<OutletId, OutletId>
) -> TractResult<TVec<OutletId>>

fn nboutputs(&self) -> TractResult<usize>

impl StatefullOp for Source

fn state(
    &self,
    _session: &mut SessionState,
    node_id: usize
) -> TractResult<Option<Box<dyn OpState>>>

fn as_stateless(&self) -> Option<&dyn StatelessOp>

impl Op for Source

fn name(&self) -> Cow<str>

fn as_typed(&self) -> Option<&dyn TypedOp>

fn incorporate(
    &self,
    _model: &InferenceModel,
    _node: &InferenceNode
) -> TractResult<Option<InferenceModelPatch>>

Early pass on inference model, after analyse, but before translation to typed network. Meant to deal with some framework idiosyncrasies that manifest with temporaries nodes that can run some form of inference but require refactoring the network before it can be evaluated.

fn declutter(
    &self,
    _model: &TypedModel,
    _node: &TypedNode
) -> TractResult<Option<TypedModelPatch>>

Declutter the op to the tract_core operator set as much as possible.

fn codegen(
    &self,
    _model: &TypedModel,
    _node: &TypedNode
) -> TractResult<Option<TypedModelPatch>>

Translate the op into the most efficient form possible for execution.

fn fuse(
    &self,
    _model: &TypedModel,
    _node: &TypedNode
) -> TractResult<Option<TypedModelPatch>>

Fuse op after codegen to deal with local optimisations.

fn cost(&self, _inputs: &[&TypedTensorInfo]) -> TractResult<TVec<(Cost, TDim)>>

Computes a cost hint of the operation.

fn nested_models(&self) -> Vec<(Cow<str>, &dyn Model)>

Nested models, with label (for audit).

fn validation(&self) -> Validation

The kind of accuracy check that should be performed on operation when testing them.

fn axes_info(
    &self,
    _model: &TypedModel,
    _node: &TypedNode
) -> TractResult<AxesInfo>

fn same_as(&self, _other: &dyn Op) -> bool

Compare two ops.

fn info(&self) -> TractResult<Vec<String>>

Short (one-line) strings giving hints on internal implementation or important configuration details to be displayed in dumps.

fn is_canonic(&self) -> bool

impl Clone for Source

fn clone(&self) -> Source

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for Source

fn fmt(&self, f: &mut Formatter) -> Result

Formats the value using the given formatter.