Struct tract_onnx::prelude::ModelPatch

pub struct ModelPatch<F, O> where
    F: Fact + Clone + 'static + Hash,
    O: Display + Debug + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,  {
    pub model: Graph<F, O>,
    pub incoming: HashMap<OutletId, OutletId, RandomState>,
    pub shunt_outlet_by: HashMap<OutletId, OutletId, RandomState>,
    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

model: Graph<F, O>

the model-like 'pagch' of nodes to add to the model

incoming: HashMap<OutletId, OutletId, RandomState>

shunt_outlet_by: HashMap<OutletId, OutletId, RandomState>

obliterate: Vec<usize>

Implementations

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

pub fn is_empty(&self) -> bool

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

Draw a tap from a preexisting node.

returns an OutletId usable in the little "patch" model

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

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

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

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

Convenience method creating a patch that replace a single operation.

pub fn fuse_with_next<IO>(
    patched_model: &Graph<F, O>,
    node: &BaseNode<F, O>,
    new_op: IO
) -> Result<ModelPatch<F, O>, TractError> where
    IO: Into<O>, 

Convenience method creating a patch that replace a single operation.

pub fn shunt_one_op(
    patched_model: &Graph<F, O>,
    node: &BaseNode<F, O>
) -> Result<ModelPatch<F, O>, TractError>

Convenience method creating a patch that shunt the given node.

pub fn single_unary_op<IO>(
    patched_model: &Graph<F, O>,
    node: &BaseNode<F, O>,
    new_op: IO
) -> Result<ModelPatch<F, O>, TractError> where
    IO: Into<O>, 

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

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

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

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

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
) -> Result<OutletId, TractError>

pub fn add_node(
    &mut self,
    name: impl Into<String>,
    op: impl Into<O>,
    output_facts: SmallVec<[F; 4]>
) -> Result<usize, TractError>

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

Connect a node outlet to a node inlet.

pub fn input_outlets(&self) -> Result<&[OutletId], TractError>

Get model inputs.

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

Change model inputs.

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

Set model inputs by the node name.

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

Get the ix-th input tensor type information.

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

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

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

Set the ix-th input tensor type information.

pub fn output_outlets(&self) -> Result<&[OutletId], TractError>

Get model outputs.

pub fn auto_outputs(&mut self) -> Result<(), TractError>

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

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

Change model outputs.

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

Set model outputs by node names.

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

Get the ix-th input tensor type information.

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

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

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

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) -> Result<usize, TractError>

pub fn node_by_name(
    &self,
    name: impl AsRef<str>
) -> Result<&BaseNode<F, O>, TractError>

Find a node by its name.

pub fn node_by_name_mut(
    &mut self,
    name: impl AsRef<str>
) -> Result<&mut BaseNode<F, O>, TractError>

Borrow mutably a node by its name.

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

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

Find a node by its id.

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

Find a node by its id.

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

Access the nodes table.

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

Access the nodes table.

pub fn node_facts(
    &self,
    id: usize
) -> Result<(SmallVec<[&F; 4]>, SmallVec<[&F; 4]>), TractError>

Get input and output tensor information for a node.

pub fn node_input_facts(
    &self,
    node_id: usize
) -> Result<SmallVec<[&F; 4]>, TractError>

Get input tensor information for a node.

pub fn node_output_facts(
    &self,
    node_id: usize
) -> Result<SmallVec<[&F; 4]>, TractError>

Get output tensor information for a node.

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

Get tensor information for a single outlet.

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

Get tensor information for a single outlet.

pub fn outlets_fact_mut(
    &mut self,
    outlets: &[OutletId]
) -> Result<SmallVec<[&mut F; 4]>, TractError>

Get multiple mutable tensor information for outlets.

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

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
) -> Result<(), TractError>

Set label for an outlet.

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

Find outlet by label.

pub fn eval_order(&self) -> Result<Vec<usize>, TractError>

Computes an evalutation order for the graph inputs and outputs

pub fn check_edges(&self) -> Result<(), TractError>

Performs a sanity check on network connections.

pub fn single_prec(
    &self,
    id: usize
) -> Result<Option<&BaseNode<F, O>>, TractError>

pub fn single_prec_at(
    &self,
    id: usize,
    count: usize
) -> Result<Option<&BaseNode<F, O>>, TractError>

pub fn single_succ_at(
    &self,
    id: usize,
    count: usize
) -> Result<Option<&BaseNode<F, O>>, TractError>

pub fn single_succ(
    &self,
    id: usize
) -> Result<Option<&BaseNode<F, O>>, TractError>

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

pub fn add_const(
    &mut self,
    name: impl Into<String>,
    v: impl IntoArcTensor
) -> Result<OutletId, TractError>

pub fn compact(&self) -> Result<Graph<F, O>, TractError>

pub fn signature(&self) -> u64

pub fn declutter(
    &self
) -> Result<Graph<TypedFact, Box<dyn TypedOp + 'static>>, TractError>

Perform declutter passes on the network.

pub fn concretize_stream_dim(
    &self,
    dim: usize
) -> Result<Graph<TypedFact, Box<dyn TypedOp + 'static>>, TractError>

pub fn invariants(&self) -> Result<Invariants, TractError>

Trait Implementations

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

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

Returns a copy 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: Debug + Fact + Clone + 'static + Hash,
    O: Debug + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash, 

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

Formats the value using the given formatter.

impl<F, O> Default for ModelPatch<F, O> where
    F: Fact + Clone + 'static + Hash,
    O: Display + Debug + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash, 

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 + Hash,
    O: Display + Debug + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash, 

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 + Hash,
    O: Display + Debug + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash, 

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

Mutably dereferences the value.