Type Definition tract_core::model::InferenceModel

type InferenceModel = Model<TensorFact>;

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

Methods

impl InferenceModel

pub fn analyse_one(&mut self, id: usize) -> TractResult<()>

Analyse one node of the graph.

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

Analyse all nodes of the graph.

Will stop on first error unless obstinate is true.

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

List OutletId with incomplete type information.

Will stop on first error unless obstinate is true.

pub fn into_typed(self) -> TractResult<TypedModel>

Attempt full analyse and conversion to TypedModel.

pub fn into_normalized(self) -> TractResult<NormalizedModel>

Attempt full analyse, decluttering and conversion to NormalizedModel.

pub 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.

Trait Implementations

impl ModelDslConst for InferenceModel

fn add_const(
    &mut self,
    name: impl Into<String>,
    v: impl IntoArcTensor
) -> TractResult<usize>

Add a constant node to the graph.

fn plug_const(
    &mut self,
    inlet: InletId,
    name: impl Into<String>,
    v: impl IntoArcTensor
) -> TractResult<()>

Add a constant node to the graph and connect its output to inlet.

impl ModelDslInfer for InferenceModel

fn add_source_default(&mut self, name: impl Into<String>) -> TractResult<usize>

Add a source with no tensor information.

fn add_node_default(
    &mut self,
    name: impl Into<String>,
    op: impl Into<Box<dyn Op>>
) -> TractResult<usize>

Add a node without tensor information.

fn chain_default(
    &mut self,
    name: impl Into<String>,
    op: impl Into<Box<dyn Op>>
) -> TractResult<usize>

Chain a node without tensor information.