[−][src]Struct tract_core::model::ModelImpl

pub struct ModelImpl<TI, O> where
    TI: Fact + Clone + 'static,
    O: Debug + Display + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static,  { /* fields omitted */ }

Main model class

Parameterized by a Fact class.

Methods

`impl<TI, O> ModelImpl<TI, O> where TI: Fact + Clone + 'static, O: Debug + Display + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static, ModelImpl<TI, O>: Model,` [src]

`pub fn add_node( &mut self, name: impl Into<String>, op: impl Into<O>, output_facts: TVec<TI> ) -> TractResult<usize>`[src]

`pub fn add_edge(&mut self, outlet: OutletId, inlet: InletId) -> TractResult<()>`[src]

Connect a node outlet to a node inlet.

`pub fn input_outlets(&self) -> TractResult<&[OutletId]>`[src]

Get model inputs.

`pub fn set_input_outlets(&mut self, inputs: &[OutletId]) -> TractResult<()>`[src]

Change model inputs.

`pub fn set_input_names( &mut self, inputs: impl IntoIterator<Item = impl AsRef<str>> ) -> TractResult<()>`[src]

Set model inputs by the node name.

`pub fn input_fact(&self, ix: usize) -> TractResult<&TI>`[src]

Get the ix-th input tensor type information.

`pub fn input_fact_mut(&mut self, ix: usize) -> TractResult<&mut TI>`[src]

Get the ix-th input tensor type information, mutably.

`pub fn set_input_fact(&mut self, input: usize, fact: TI) -> TractResult<()>`[src]

Set the ix-th input tensor type information.

`pub fn output_outlets(&self) -> TractResult<&[OutletId]>`[src]

Get model outputs.

`pub fn auto_outputs(&mut self) -> TractResult<()>`[src]

Guess outputs from the topology: node or nodes with no successors.

`pub fn set_output_outlets(&mut self, outputs: &[OutletId]) -> TractResult<()>`[src]

Change model outputs.

`pub fn set_output_names( &mut self, outputs: impl IntoIterator<Item = impl AsRef<str>> ) -> TractResult<()>`[src]

Set model outputs by node names.

`pub fn output_fact(&self, ix: usize) -> TractResult<&TI>`[src]

Get the ix-th input tensor type information.

`pub fn output_fact_mut(&mut self, ix: usize) -> TractResult<&mut TI>`[src]

Get the ix-th input tensor type information, mutably.

`pub fn set_output_fact(&mut self, output: usize, fact: TI) -> TractResult<()>`[src]

Set the ix-th output tensor type information.

`pub fn node_names(&self) -> impl Iterator<Item = &str>`[src]

Iterate over all node names.

`pub fn node_by_name<S: AsRef<str>>( &self, name: S ) -> TractResult<&BaseNode<TI, O>>`[src]

Find a node by its name.

`pub fn node_by_name_mut( &mut self, name: &str ) -> TractResult<&mut BaseNode<TI, O>>`[src]

Borrow mutably a node by its name.

`pub fn rename_node(&mut self, id: usize, name: &str) -> TractResult<()>`[src]

`pub fn node(&self, id: usize) -> &BaseNode<TI, O>`[src]

Find a node by its id.

`pub fn node_mut(&mut self, id: usize) -> &mut BaseNode<TI, O>`[src]

Find a node by its id.

`pub fn nodes(&self) -> &[BaseNode<TI, O>]`[src]

Access the nodes table.

`pub fn nodes_mut(&mut self) -> &mut [BaseNode<TI, O>]`[src]

Access the nodes table.

`pub fn node_facts(&self, id: usize) -> TractResult<(TVec<&TI>, TVec<&TI>)>`[src]

Get input and output tensor information for a node.

`pub fn node_input_facts(&self, node_id: usize) -> TractResult<TVec<&TI>>`[src]

Get input tensor information for a node.

`pub fn node_output_facts(&self, node_id: usize) -> TractResult<TVec<&TI>>`[src]

Get output tensor information for a node.

`pub fn outlet_fact(&self, outlet: OutletId) -> TractResult<&TI>`[src]

Get tensor information for a single outlet.

`pub fn outlet_fact_mut(&mut self, outlet: OutletId) -> TractResult<&mut TI>`[src]

Get tensor information for a single outlet.

`pub fn outlets_fact_mut( &mut self, outlets: &[OutletId] ) -> TractResult<TVec<&mut TI>>`[src]

Get multiple mutable tensor information for outlets.

`pub fn set_outlet_fact(&mut self, outlet: OutletId, fact: TI) -> TractResult<()>`[src]

Set tensor information for a single outlet.

`pub fn outlet_label(&self, outlet: OutletId) -> Option<&str>`[src]

Get label for an outlet.

`pub fn set_outlet_label(&mut self, outlet: OutletId, label: String)`[src]

Set label for an outlet.

`pub fn find_outlet_label(&self, label: &str) -> Option<OutletId>`[src]

Find outlet by label.

`pub fn eval_order(&self) -> TractResult<Vec<usize>>`[src]

Computes an evalutation order for the graph inputs and outputs

`pub fn check_edges(&self) -> TractResult<()>`[src]

Performs a sanity check on network connections.

Trait Implementations

`impl<TI, O> ModelDsl<TI, O> for ModelImpl<TI, O> where TI: Fact + Clone + 'static, O: Debug + Display + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static,` [src]

`fn single_prec(&self, id: usize) -> TractResult<Option<&BaseNode<TI, O>>>`[src]

`fn single_prec_at( &self, id: usize, count: usize ) -> TractResult<Option<&BaseNode<TI, O>>>`[src]

`fn single_succ_at( &self, id: usize, count: usize ) -> TractResult<Option<&BaseNode<TI, O>>>`[src]

`fn single_succ(&self, id: usize) -> TractResult<Option<&BaseNode<TI, O>>>`[src]

`impl<TI: Fact + Clone + 'static, O, E> ModelDslConst for ModelImpl<TI, O> where TractError: From<E>, TI: Fact + Clone + 'static + TryFrom<InferenceFact, Error = E>, O: Debug + Display + From<Const> + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static,` [src]

`fn add_const( &mut self, name: impl Into<String>, v: impl IntoArcTensor ) -> TractResult<OutletId>`[src]

`impl<TI, O> Model for ModelImpl<TI, O> where TI: Fact + Clone + 'static, O: Debug + Display + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static,` [src]

`fn node_id_by_name(&self, name: &str) -> TractResult<usize>`[src]

`fn node_name(&self, id: usize) -> &str`[src]

`fn node_inputs(&self, id: usize) -> &[OutletId]`[src]

`fn node_output_count(&self, id: usize) -> usize`[src]

`fn node_control_inputs(&self, id: usize) -> &[usize]`[src]

`fn nodes_len(&self) -> usize`[src]

`fn node_format(&self, id: usize) -> String`[src]

`fn eval_order(&self) -> TractResult<Vec<usize>>`[src]

`fn eval_order_for_io( &self, inputs: &[usize], outputs: &[usize] ) -> TractResult<Vec<usize>>`[src]

`fn input_outlets(&self) -> &[OutletId]`[src]

`fn output_outlets(&self) -> &[OutletId]`[src]

`fn node_op(&self, id: usize) -> &dyn Op`[src]

`fn outlet_tensorfact(&self, outlet: OutletId) -> InferenceFact`[src]

`fn outlet_fact_format(&self, outlet: OutletId) -> String`[src]

`fn outlet_label(&self, id: OutletId) -> Option<&str>`[src]

`fn outlet_successors(&self, outlet: OutletId) -> &[InletId]`[src]

`impl<TI: Clone, O: Clone> Clone for ModelImpl<TI, O> where TI: Fact + Clone + 'static, O: Debug + Display + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static,` [src]

`fn clone(&self) -> ModelImpl<TI, O>`[src]

`fn clone_from(&mut self, source: &Self)`1.0.0[src]

`impl<TI, O> Default for ModelImpl<TI, O> where TI: Fact + Clone + 'static, O: Debug + Display + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static,` [src]

`fn default() -> ModelImpl<TI, O>`[src]

`impl<TI: Debug, O: Debug> Debug for ModelImpl<TI, O> where TI: Fact + Clone + 'static, O: Debug + Display + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static,` [src]

`fn fmt(&self, f: &mut Formatter) -> Result`[src]

Auto Trait Implementations

`impl<TI, O> Send for ModelImpl<TI, O> where O: Send,`

`impl<TI, O> Sync for ModelImpl<TI, O> where O: Sync,`

`impl<TI, O> Unpin for ModelImpl<TI, O> where O: Unpin, TI: Unpin,`

`impl<TI, O> UnwindSafe for ModelImpl<TI, O> where O: UnwindSafe, TI: RefUnwindSafe + UnwindSafe,`

`impl<TI, O> RefUnwindSafe for ModelImpl<TI, O> where O: RefUnwindSafe, TI: RefUnwindSafe,`

Blanket Implementations

`impl<T, U> Into for T where U: From<T>,` [src]

`fn into(self) -> U`[src]

`impl<T> From<T> for T`[src]

`fn from(t: T) -> T`[src]

`impl<T> ToOwned for T where T: Clone,` [src]

`type Owned = T`

The resulting type after obtaining ownership.

`fn to_owned(&self) -> T`[src]

`fn clone_into(&self, target: &mut T)`[src]

`impl<T, U> TryFrom for T where U: Into<T>,` [src]

`type Error = Infallible`

The type returned in the event of a conversion error.

`fn try_from(value: U) -> Result<T, <T as TryFrom>::Error>`[src]

`impl<T, U> TryInto for T where U: TryFrom<T>,` [src]

`type Error = >::Error`

The type returned in the event of a conversion error.

[−][src]Struct tract_core::model::ModelImpl

Methods

impl<TI, O> ModelImpl<TI, O> where TI: Fact + Clone + 'static, O: Debug + Display + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static, ModelImpl<TI, O>: Model, [src]

pub fn add_node( &mut self, name: impl Into<String>, op: impl Into<O>, output_facts: TVec<TI>) -> TractResult<usize>[src]

pub fn add_edge(&mut self, outlet: OutletId, inlet: InletId) -> TractResult<()>[src]

pub fn input_outlets(&self) -> TractResult<&[OutletId]>[src]

pub fn set_input_outlets(&mut self, inputs: &[OutletId]) -> TractResult<()>[src]

pub fn set_input_names( &mut self, inputs: impl IntoIterator<Item = impl AsRef<str>>) -> TractResult<()>[src]

pub fn input_fact(&self, ix: usize) -> TractResult<&TI>[src]

pub fn input_fact_mut(&mut self, ix: usize) -> TractResult<&mut TI>[src]

pub fn set_input_fact(&mut self, input: usize, fact: TI) -> TractResult<()>[src]

pub fn output_outlets(&self) -> TractResult<&[OutletId]>[src]

pub fn auto_outputs(&mut self) -> TractResult<()>[src]

pub fn set_output_outlets(&mut self, outputs: &[OutletId]) -> TractResult<()>[src]

pub fn set_output_names( &mut self, outputs: impl IntoIterator<Item = impl AsRef<str>>) -> TractResult<()>[src]

pub fn output_fact(&self, ix: usize) -> TractResult<&TI>[src]

pub fn output_fact_mut(&mut self, ix: usize) -> TractResult<&mut TI>[src]

pub fn set_output_fact(&mut self, output: usize, fact: TI) -> TractResult<()>[src]

pub fn node_names(&self) -> impl Iterator<Item = &str>[src]

pub fn node_by_name<S: AsRef<str>>( &self, name: S) -> TractResult<&BaseNode<TI, O>>[src]

pub fn node_by_name_mut( &mut self, name: &str) -> TractResult<&mut BaseNode<TI, O>>[src]

pub fn rename_node(&mut self, id: usize, name: &str) -> TractResult<()>[src]

pub fn node(&self, id: usize) -> &BaseNode<TI, O>[src]

pub fn node_mut(&mut self, id: usize) -> &mut BaseNode<TI, O>[src]

pub fn nodes(&self) -> &[BaseNode<TI, O>][src]

pub fn nodes_mut(&mut self) -> &mut [BaseNode<TI, O>][src]

pub fn node_facts(&self, id: usize) -> TractResult<(TVec<&TI>, TVec<&TI>)>[src]

pub fn node_input_facts(&self, node_id: usize) -> TractResult<TVec<&TI>>[src]

pub fn node_output_facts(&self, node_id: usize) -> TractResult<TVec<&TI>>[src]

pub fn outlet_fact(&self, outlet: OutletId) -> TractResult<&TI>[src]

pub fn outlet_fact_mut(&mut self, outlet: OutletId) -> TractResult<&mut TI>[src]

pub fn outlets_fact_mut( &mut self, outlets: &[OutletId]) -> TractResult<TVec<&mut TI>>[src]

pub fn set_outlet_fact(&mut self, outlet: OutletId, fact: TI) -> TractResult<()>[src]

pub fn outlet_label(&self, outlet: OutletId) -> Option<&str>[src]

pub fn set_outlet_label(&mut self, outlet: OutletId, label: String)[src]

pub fn find_outlet_label(&self, label: &str) -> Option<OutletId>[src]

pub fn eval_order(&self) -> TractResult<Vec<usize>>[src]

pub fn check_edges(&self) -> TractResult<()>[src]

Trait Implementations

impl<TI, O> ModelDsl<TI, O> for ModelImpl<TI, O> where TI: Fact + Clone + 'static, O: Debug + Display + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static, [src]

fn single_prec(&self, id: usize) -> TractResult<Option<&BaseNode<TI, O>>>[src]

fn single_prec_at( &self, id: usize, count: usize) -> TractResult<Option<&BaseNode<TI, O>>>[src]

fn single_succ_at( &self, id: usize, count: usize) -> TractResult<Option<&BaseNode<TI, O>>>[src]

fn single_succ(&self, id: usize) -> TractResult<Option<&BaseNode<TI, O>>>[src]

impl<TI: Fact + Clone + 'static, O, E> ModelDslConst for ModelImpl<TI, O> where TractError: From<E>, TI: Fact + Clone + 'static + TryFrom<InferenceFact, Error = E>, O: Debug + Display + From<Const> + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static, [src]

fn add_const( &mut self, name: impl Into<String>, v: impl IntoArcTensor) -> TractResult<OutletId>[src]

impl<TI, O> Model for ModelImpl<TI, O> where TI: Fact + Clone + 'static, O: Debug + Display + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static, [src]

fn node_id_by_name(&self, name: &str) -> TractResult<usize>[src]

fn node_name(&self, id: usize) -> &str[src]

fn node_inputs(&self, id: usize) -> &[OutletId][src]

fn node_output_count(&self, id: usize) -> usize[src]

fn node_control_inputs(&self, id: usize) -> &[usize][src]

fn nodes_len(&self) -> usize[src]

fn node_format(&self, id: usize) -> String[src]

fn eval_order(&self) -> TractResult<Vec<usize>>[src]

fn eval_order_for_io( &self, inputs: &[usize], outputs: &[usize]) -> TractResult<Vec<usize>>[src]

fn input_outlets(&self) -> &[OutletId][src]

fn output_outlets(&self) -> &[OutletId][src]

fn node_op(&self, id: usize) -> &dyn Op[src]

fn outlet_tensorfact(&self, outlet: OutletId) -> InferenceFact[src]

fn outlet_fact_format(&self, outlet: OutletId) -> String[src]

fn outlet_label(&self, id: OutletId) -> Option<&str>[src]

fn outlet_successors(&self, outlet: OutletId) -> &[InletId][src]

impl<TI: Clone, O: Clone> Clone for ModelImpl<TI, O> where TI: Fact + Clone + 'static, O: Debug + Display + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static, [src]

fn clone(&self) -> ModelImpl<TI, O>[src]

fn clone_from(&mut self, source: &Self)1.0.0[src]

impl<TI, O> Default for ModelImpl<TI, O> where TI: Fact + Clone + 'static, O: Debug + Display + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static, [src]

fn default() -> ModelImpl<TI, O>[src]

impl<TI: Debug, O: Debug> Debug for ModelImpl<TI, O> where TI: Fact + Clone + 'static, O: Debug + Display + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static, [src]

fn fmt(&self, f: &mut Formatter) -> Result[src]