Struct ModelPatch 

pub struct ModelPatch<F, O>
where
    F: Fact + Clone + 'static,
    O: Display + Debug + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static,
{
    pub context: Vec<String>,
    pub dont_apply_twice: Option<String>,
    pub model: Graph<F, O>,
    pub inputs: HashMap<usize, usize>,
    pub taps: HashMap<OutletId, OutletId>,
    pub shunts: HashMap<OutletId, OutletId>,
    pub obliterate: Vec<usize>,
}

A change to apply to a model.

Actually structured around a model that represent the new nodes to be inserted, plus information about how to connect these new nodes to the pre-existing graph.

Fields

context: Vec<String>

patch label for auditing and debugging

dont_apply_twice: Option<String>

optimizer will ignore this patch in node to node loop if it was already encountered

model: Graph<F, O>

the model-like ‘patch’ of nodes to add to the model

inputs: HashMap<usize, usize>

map of replaced inputs (patch node id to model node id)

taps: HashMap<OutletId, OutletId>

map of patch inputs to model wires

shunts: HashMap<OutletId, OutletId>

map of old model wires to be replaced by wires from the patch

obliterate: Vec<usize>

operations to discard from the model

Implementations

impl<F, O> ModelPatch<F, O>

where F: Fact + Clone + 'static, O: Display + Debug + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static, Graph<F, O>: SpecialOps<F, O>,

pub fn new(s: impl Into<String>) -> Self

pub fn push_context(&mut self, s: impl Into<String>)

pub fn with_context(self, s: impl Into<String>) -> Self

pub fn is_empty(&self) -> bool

pub fn tap_model( &mut self, model: &Graph<F, O>, outlet: OutletId, ) -> TractResult<OutletId>

Draw a tap from a preexisting node.

returns an OutletId usable in the little “patch” model

pub fn taps<'a>( &mut self, model: &Graph<F, O>, outlets: impl IntoIterator<Item = &'a OutletId>, ) -> TractResult<TVec<OutletId>>

Draw taps from a preexisting node.

returns an OutletId usable in the little “patch” model

pub unsafe fn shunt_outside_unchecked( &mut self, outlet: OutletId, by: OutletId, ) -> TractResult<()>

pub fn shunt_outside( &mut self, model: &Graph<F, O>, outlet: OutletId, by: OutletId, ) -> TractResult<()>

Replace an Outlet in the target model by one from the patch.

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

pub fn replace_single_op<IO: Into<O>>( patched_model: &Graph<F, O>, node: &Node<F, O>, inputs: &[OutletId], new_op: IO, ) -> TractResult<ModelPatch<F, O>>

Convenience method creating a patch that replaces a single operation.

pub fn fuse_with_next<IO: Into<O>>( patched_model: &Graph<F, O>, node: &Node<F, O>, new_op: IO, ) -> TractResult<ModelPatch<F, O>>

Convenience method creating a patch that fuses an op with the next one.

pub fn shunt_one_op( patched_model: &Graph<F, O>, node: &Node<F, O>, ) -> TractResult<Option<ModelPatch<F, O>>>

Convenience method creating a patch that shunts the given node.

pub fn rewire( patched_model: &Graph<F, O>, from: &[OutletId], to: &[OutletId], wiring: &dyn Fn(&mut Self, &[OutletId]) -> TractResult<TVec<OutletId>>, ) -> TractResult<ModelPatch<F, O>>

pub fn single_unary_op<IO: Into<O>>( patched_model: &Graph<F, O>, node: &Node<F, O>, new_op: IO, ) -> TractResult<ModelPatch<F, O>>

Convenience method creating a patch that replace a single unary operation.

pub fn intercept<IO: Into<O>>( patched_model: &Graph<F, O>, outlet: OutletId, name: impl Into<String>, new_op: IO, fact: F, ) -> TractResult<ModelPatch<F, O>>

Convenience method creating a patch that insert an unary op on an outlet.

pub fn wire_node( &mut self, name: impl Into<String>, op: impl Into<O>, inputs: &[OutletId], ) -> TractResult<TVec<OutletId>>

pub fn apply(self, target: &mut Graph<F, O>) -> TractResult<()>

Apply all changes in the patch to the target model.

Methods from Deref<Target = Graph<F, O>>

pub fn add_source( &mut self, name: impl Into<String>, fact: F, ) -> TractResult<OutletId>

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

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

Connect a node outlet to a node inlet.

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

Get model inputs.

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

Change model inputs.

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

Set model inputs by the node name.

pub fn input_fact(&self, ix: usize) -> TractResult<&F>

Get the ix-th input tensor type information.

pub fn input_fact_mut(&mut self, ix: usize) -> TractResult<&mut F>

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

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

Set the ix-th input tensor type information.

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

Get model outputs.

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

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

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

Change model outputs.

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

Set model outputs by node names.

pub fn output_fact(&self, ix: usize) -> TractResult<&F>

Get the ix-th input tensor type information.

pub fn output_fact_mut(&mut self, ix: usize) -> TractResult<&mut F>

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

pub fn set_output_fact(&mut self, output: usize, fact: F) -> TractResult<()>

Set the ix-th output tensor type information.

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

Iterate over all node names.

pub fn node_id_by_name(&self, name: &str) -> TractResult<usize>

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

Find a node by its name.

pub fn node_by_name_mut( &mut self, name: impl AsRef<str>, ) -> TractResult<&mut Node<F, O>>

Borrow mutably a node by its name.

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

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

Find a node by its id.

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

Find a node by its id.

pub fn nodes(&self) -> &[Node<F, O>]

Access the nodes table.

pub fn nodes_mut(&mut self) -> &mut [Node<F, O>]

Access the nodes table.

pub fn node_facts(&self, id: usize) -> TractResult<(TVec<&F>, TVec<&F>)>

Get input and output tensor information for a node.

pub fn node_input_facts(&self, node_id: usize) -> TractResult<TVec<&F>>

Get input tensor information for a node.

pub fn node_output_facts(&self, node_id: usize) -> TractResult<TVec<&F>>

Get output tensor information for a node.

pub fn outlet_fact(&self, outlet: OutletId) -> TractResult<&F>

Get tensor information for a single outlet.

pub fn outlet_fact_mut(&mut self, outlet: OutletId) -> TractResult<&mut F>

Get tensor information for a single outlet.

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

Get multiple mutable tensor information for outlets.

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

Set tensor information for a single outlet.

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

Get label for an outlet.

pub fn set_outlet_label( &mut self, outlet: OutletId, label: String, ) -> TractResult<()>

Set label for an outlet.

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

Find outlet by label.

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

Computes an evalutation order for the graph inputs and outputs

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

Computes an evalutation order for the graph inputs and outputs. This order will minimize temporary buffers.

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

pub fn eval_tmp_memory_usage<Flushable>( &self, order: &[usize], flushable: Flushable, ) -> TractResult<TVec<(usize, TDim)>>

where Flushable: Fn(&Node<F, O>) -> bool,

Evaluate temporary memory usage with its related node at each step of the given order.

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

pub fn linear_prec(&self, id: usize) -> TractResult<Option<&Node<F, O>>>

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

pub fn all_prec(&self, id: usize) -> TractResult<Option<TVec<&Node<F, O>>>>

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

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

pub fn linear_succ(&self, id: usize) -> TractResult<Option<&Node<F, O>>>

linear_succ is only intended for optimisation of simple operators with 1 output, and only 1 output successors (successor with only 1 input)

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

pub fn all_succ(&self, id: usize) -> TractResult<Option<TVec<&Node<F, O>>>>

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

pub fn sym(&self, s: &str) -> Symbol

retrieve of create a symbol

pub fn new_sym_with_prefix(&self, prefix: &str) -> Symbol

create a new symbol with the prefix

pub fn unique_name<'n>(&self, prefix: impl Into<Cow<'n, str>>) -> Cow<'n, str>

generates a name for a new node in the model that will not conflict (by suffixing with a dot and number)

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

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

Trait Implementations

impl<F, O> Clone for ModelPatch<F, O>

where F: Fact + Clone + 'static + Clone, O: Display + Debug + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static + Clone,

fn clone(&self) -> ModelPatch<F, O>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<F, O> Debug for ModelPatch<F, O>

where F: Fact + Clone + 'static + Debug, O: Display + Debug + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static + Debug,

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

Formats the value using the given formatter.

impl<F, O> Default for ModelPatch<F, O>

where F: Fact + Clone + 'static, O: Display + Debug + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static,

fn default() -> ModelPatch<F, O>

Returns the “default value” for a type.

impl<F, O> Deref for ModelPatch<F, O>

where F: Fact + Clone + 'static, O: Display + Debug + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static,

type Target = Graph<F, O>

The resulting type after dereferencing.

fn deref(&self) -> &Graph<F, O>

Dereferences the value.

impl<F, O> DerefMut for ModelPatch<F, O>

where F: Fact + Clone + 'static, O: Display + Debug + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static,

fn deref_mut(&mut self) -> &mut Graph<F, O>

Mutably dereferences the value.