[−][src]Trait tract_hir::prelude::InferenceModelExt

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

    fn incorporate(self) -> TractResult<InferenceModel>;

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

    fn eliminate_dead_branches(self) -> TractResult<InferenceModel>;

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

    fn into_optimized(self) -> TractResult<TypedModel>;
}

Required methods

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_optimized(self) -> TractResult<TypedModel>

Attempt full analyse, decluttering and mapping to optimized operations.

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

Loading content...

Implementors

impl InferenceModelExt for InferenceModel[src]

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

Analyse all nodes of the graph.

Will stop on first error unless obstinate is true.

fn incorporate(self) -> TractResult<InferenceModel>[src]

Perform early transformation before going typed.

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

List OutletId with incomplete type information.

Will stop on first error unless obstinate is true.

fn eliminate_dead_branches(self) -> TractResult<InferenceModel>[src]

Eliminate seemingly dead branches of the graph.

This may break stateful networks.

fn into_typed(self) -> TractResult<TypedModel>[src]

Attempt full analyse and conversion to TypedModel.

fn into_optimized(self) -> TractResult<TypedModel>[src]

Attempt full analyse, decluttering and mapping to optimized operations.

This is meant for "simple" networks, where no special model transformation needs to happen. Aternaltively, use to_typed() and manipulate the TypedModel for more control.

Loading content...