[−][src]Type Definition tract_hir::infer::InferenceModel
type InferenceModel = Graph<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
[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.
impl SpecialOps<InferenceFact, Box<dyn InferenceOp + 'static>> for InferenceModel
[src]
fn is_source(op: &Box<dyn InferenceOp>) -> bool
[src]
fn create_dummy(&self) -> Box<dyn InferenceOp>ⓘNotable traits for Box<W>
impl<W> Write for Box<W> where
W: Write + ?Sized, impl<R> Read for Box<R> where
R: Read + ?Sized, impl<I> Iterator for Box<I> where
I: Iterator + ?Sized, type Item = <I as Iterator>::Item;impl<F> Future for Box<F> where
F: Unpin + Future + ?Sized, type Output = <F as Future>::Output;
[src]
Notable traits for Box<W>
impl<W> Write for Box<W> where
W: Write + ?Sized, impl<R> Read for Box<R> where
R: Read + ?Sized, impl<I> Iterator for Box<I> where
I: Iterator + ?Sized, type Item = <I as Iterator>::Item;impl<F> Future for Box<F> where
F: Unpin + Future + ?Sized, type Output = <F as Future>::Output;
fn create_source(&self, _fact: InferenceFact) -> Box<dyn InferenceOp>ⓘNotable traits for Box<W>
impl<W> Write for Box<W> where
W: Write + ?Sized, impl<R> Read for Box<R> where
R: Read + ?Sized, impl<I> Iterator for Box<I> where
I: Iterator + ?Sized, type Item = <I as Iterator>::Item;impl<F> Future for Box<F> where
F: Unpin + Future + ?Sized, type Output = <F as Future>::Output;
[src]
Notable traits for Box<W>
impl<W> Write for Box<W> where
W: Write + ?Sized, impl<R> Read for Box<R> where
R: Read + ?Sized, impl<I> Iterator for Box<I> where
I: Iterator + ?Sized, type Item = <I as Iterator>::Item;impl<F> Future for Box<F> where
F: Unpin + Future + ?Sized, type Output = <F as Future>::Output;
fn wire_node(
&mut self,
name: impl Into<String>,
op: impl Into<Box<dyn InferenceOp>>,
inputs: &[OutletId]
) -> TractResult<TVec<OutletId>>
[src]
&mut self,
name: impl Into<String>,
op: impl Into<Box<dyn InferenceOp>>,
inputs: &[OutletId]
) -> TractResult<TVec<OutletId>>