Type Definition tract_hir::infer::InferenceModel

type InferenceModel = ModelImpl<InferenceFact, Box<dyn InferenceOp>>;

A model with partially types and shapes, as produced by parsing ONNX or Tensorflow graphs.

Trait Implementations

impl InferenceModelExt for InferenceModel

fn analyse(&mut self, obstinate: bool) -> TractResult<bool>

Analyse all nodes of the graph.

Will stop on first error unless obstinate is true.

fn incorporate(self) -> TractResult<InferenceModel>

Perform early transformation before going typed.

fn missing_type_shape(&self) -> TractResult<Vec<OutletId>>

List OutletId with incomplete type information.

Will stop on first error unless obstinate is true.

fn eliminate_dead_branches(self) -> TractResult<InferenceModel>

Eliminate seemingly dead branches of the graph.

This may break stateful networks.

fn into_typed(self) -> TractResult<TypedModel>

Attempt full analyse and conversion to TypedModel.

fn into_normalized(self) -> TractResult<NormalizedModel>

Attempt full analyse, decluttering and conversion to NormalizedModel.

fn into_optimized(self) -> TractResult<TypedModel>

Attempt full analyse, decluttering and mapping to optimized operations.

This will work even if the network can not be normalized.

impl ModelSpecialOps<InferenceFact, Box<dyn InferenceOp + 'static>> for InferenceModel

fn add_source(
    &mut self,
    name: impl Into<String>,
    fact: InferenceFact
) -> TractResult<OutletId>

Adds a source op to the network.

fn is_source(op: &dyn Op) -> bool

fn create_dummy(&self) -> Box<dyn InferenceOp>

impl ModelWireNode<InferenceFact, Box<dyn InferenceOp + 'static>> for InferenceModel

fn wire_node(
    &mut self,
    name: impl Into<String>,
    op: impl Into<Box<dyn InferenceOp>>,
    inputs: &[OutletId]
) -> TractResult<TVec<OutletId>>