Struct If 

pub struct If {
    pub then_body: InferenceModel,
    pub else_body: InferenceModel,
    /* private fields */
}

Fields

then_body: InferenceModel

else_body: InferenceModel

Implementations

impl If

pub fn new( then_body: InferenceModel, then_input_mapping: Vec<usize>, else_body: InferenceModel, else_input_mapping: Vec<usize>, ) -> Self

Constructs a new If.

Trait Implementations

impl Clone for If

fn clone(&self) -> If

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for If

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

Formats the value using the given formatter.

impl EvalOp for If

fn is_stateless(&self) -> bool

fn eval(&self, inputs: TVec<TValue>) -> TractResult<TVec<TValue>>

fn eval_with_session( &self, node_id: usize, session: &SessionState, inputs: SmallVec<[TValue; 4]>, ) -> Result<SmallVec<[TValue; 4]>, Error>

fn state( &self, session: &mut SessionState, node_id: usize, ) -> Result<Option<Box<dyn OpState>>, Error>

impl InferenceOp for If

fn infer_facts( &mut self, inputs: TVec<&InferenceFact>, outputs: TVec<&InferenceFact>, observed: TVec<&InferenceFact>, ) -> TractResult<(TVec<InferenceFact>, TVec<InferenceFact>, TVec<InferenceFact>)>

Infer properties about inputs and output tensors. This method does not need to deal with the “trivial” stateless op with fully determined inputs cases.

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

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

Called during translation to TypedModel.

fn as_op(&self) -> &dyn Op

Reinterpret the InferenceOp as an Op.

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

Reinterpret the InferenceOp as an Op, mutably.

fn infer( &mut self, inputs: SmallVec<[&InferenceFact; 4]>, outputs: SmallVec<[&InferenceFact; 4]>, observed: SmallVec<[&InferenceFact; 4]>, ) -> Result<(SmallVec<[InferenceFact; 4]>, SmallVec<[InferenceFact; 4]>, SmallVec<[InferenceFact; 4]>), Error>

Infers properties about the input and output tensors.

fn observe_outlets( &self, _model: &Graph<InferenceFact, Box<dyn InferenceOp>>, _node: &Node<InferenceFact, Box<dyn InferenceOp>>, ) -> Result<Vec<OutletId>, Error>

Allow an op to specify a supplementary list of outlets facts that will trigger inference again.

fn incorporate( &self, model: &Graph<InferenceFact, Box<dyn InferenceOp>>, node: &Node<InferenceFact, Box<dyn InferenceOp>>, ) -> Result<Option<ModelPatch<InferenceFact, Box<dyn InferenceOp>>>, Error>

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.

impl Op for If

fn name(&self) -> StaticName

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

fn validation(&self) -> Validation

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

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

Compare two ops.

fn info(&self) -> Result<Vec<String>, Error>

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