Struct tract_core::model::Model

pub struct Model { /* fields omitted */ }

Model is Tract workhouse.

Methods

impl Model

pub fn with_context(self, ctx: Arc<dyn Context>) -> Model

pub fn add_node(&mut self, name: String, op: Box<dyn Op>) -> TractResult<usize>

pub fn clear_inputs(&mut self, node: usize) -> TractResult<()>

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

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

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

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

pub fn set_fact(     &mut self,     outlet: OutletId,     fact: TensorFact ) -> TractResult<()>

pub fn set_input_fact(     &mut self,     input: usize,     fact: TensorFact ) -> TractResult<()>

pub fn facts(     &self,     id: usize ) -> TractResult<(TVec<&TensorFact>, TVec<&TensorFact>)>

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

pub fn analyse(&mut self) -> TractResult<()>

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

pub fn into_optimized(self) -> TractResult<Model>

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

pub fn node_by_name(&self, name: &str) -> TractResult<&Node>

pub fn node_names(&self) -> Vec<&str>

pub fn node(&self, id: usize) -> &Node

pub fn node_mut(&mut self, id: usize) -> &mut Node

pub fn nodes(&self) -> &[Node]

pub fn mut_nodes(&mut self) -> &mut [Node]

pub fn fact(&self, outlet: OutletId) -> TractResult<&TensorFact>

pub fn inputs_fact(&self, ix: usize) -> TractResult<&TensorFact>

pub fn input_fact(&self) -> TractResult<&TensorFact>

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

pub fn outputs_fact(&self, ix: usize) -> TractResult<&TensorFact>

pub fn output_fact(&self) -> TractResult<&TensorFact>

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

pub fn into_arc(self) -> Arc<Model>

pub fn check_edges(&self) -> TractResult<()>

Trait Implementations

impl ModelDsl for Model

fn add_source<S: AsRef<str>>(&mut self, name: S) -> TractResult<usize>

fn add_source_fact<S: AsRef<str>>(     &mut self,     name: S,     fact: TensorFact ) -> TractResult<usize>

fn add_const<S: AsRef<str>>(     &mut self,     name: S,     v: SharedTensor ) -> TractResult<usize>

fn chain<S: AsRef<str>>(     &mut self,     name: S,     op: Box<dyn Op> ) -> TractResult<usize>

fn single_prec(&self, id: usize) -> TractResult<Option<&Node>>

fn single_prec_at(&self, id: usize, count: usize) -> TractResult<Option<&Node>>

fn single_succ_at(&self, id: usize, count: usize) -> TractResult<Option<&Node>>

fn single_succ(&self, id: usize) -> TractResult<Option<&Node>>

fn tap_and_chain<S: AsRef<str>>(     &mut self,     tap: OutletId,     name: S,     op: Box<dyn Op> ) -> TractResult<usize>

fn replace_nodes(     &mut self,     node: usize,     before: usize,     after: usize,     nodes: Vec<(String, Box<dyn Op>)> ) -> TractResult<()>

fn unlink_node(&mut self, node: usize) -> TractResult<()>

impl Default for Model

fn default() -> Model

Returns the "default value" for a type.

impl Clone for Model

fn clone(&self) -> Model

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for Model

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

Formats the value using the given formatter.