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

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

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>
Expand description

patch label for auditing and debugging

dont_apply_twice: Option<String>
Expand description

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

model: Graph<F, O>
Expand description

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

inputs: HashMap<usize, usize>
Expand description

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

incoming: HashMap<OutletId, OutletId>
Expand description

map of patch inputs to model wires

shunt_outlet_by: HashMap<OutletId, OutletId>
Expand description

map of old wires to be replaced by new wires

obliterate: Vec<usize>
Expand description

operations to discard from the model

Implementations

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

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

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

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

pub fn is_empty(&self) -> bool[src]

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

Draw a tap 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<()>[src]

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

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

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

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>>[src]

Convenience method creating a patch that replace 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>>[src]

Convenience method creating a patch that replace a single operation.

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

Convenience method creating a patch that shunt the given node.

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

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>>[src]

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

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

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>[src]

pub fn add_node(
    &mut self,
    name: impl Into<String>,
    op: impl Into<O>,
    output_facts: TVec<F>
) -> 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<&F>[src]

Get the ix-th input tensor type information.

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

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

pub fn set_input_fact(&mut self, input: usize, fact: F) -> 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<&F>[src]

Get the ix-th input tensor type information.

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

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

pub fn set_output_fact(&mut self, output: usize, fact: F) -> 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_id_by_name(&self, name: &str) -> TractResult<usize>[src]

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

Find a node by its name.

pub fn node_by_name_mut(
    &mut self,
    name: impl AsRef<str>
) -> TractResult<&mut Node<F, 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) -> &Node<F, O>[src]

Find a node by its id.

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

Find a node by its id.

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

Notable traits for &'_ [u8]

impl<'_> Read for &'_ [u8]impl<'_> Write for &'_ mut [u8]
[src]

Access the nodes table.

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

Notable traits for &'_ [u8]

impl<'_> Read for &'_ [u8]impl<'_> Write for &'_ mut [u8]
[src]

Access the nodes table.

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

Get input and output tensor information for a node.

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

Get input tensor information for a node.

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

Get output tensor information for a node.

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

Get tensor information for a single outlet.

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

Get tensor information for a single outlet.

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

Get multiple mutable tensor information for outlets.

pub fn set_outlet_fact(&mut self, outlet: OutletId, fact: F) -> 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
) -> TractResult<()>[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.

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

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

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

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

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

Notable traits for &'_ [u8]

impl<'_> Read for &'_ [u8]impl<'_> Write for &'_ mut [u8]
[src]

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

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

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

pub fn signature(&self) -> u64[src]

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

Perform declutter passes on the network.

pub fn concretize_dims(&self, values: &SymbolValues) -> TractResult<TypedModel>[src]

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

Trait Implementations

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

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

Returns a copy of the value. Read more

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

Performs copy-assignment from source. Read more

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

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

Formats the value using the given formatter. Read more

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

fn default() -> ModelPatch<F, O>[src]

Returns the “default value” for a type. Read more

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

type Target = Graph<F, O>

The resulting type after dereferencing.

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

Dereferences the value.

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

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

Mutably dereferences the value.

Auto Trait Implementations

impl<F, O> RefUnwindSafe for ModelPatch<F, O> where
    F: RefUnwindSafe,
    O: RefUnwindSafe, 

impl<F, O> Send for ModelPatch<F, O> where
    O: Send, 

impl<F, O> Sync for ModelPatch<F, O> where
    O: Sync, 

impl<F, O> Unpin for ModelPatch<F, O> where
    F: Unpin,
    O: Unpin, 

impl<F, O> UnwindSafe for ModelPatch<F, O> where
    F: RefUnwindSafe + UnwindSafe,
    O: UnwindSafe, 

Blanket Implementations

impl<T> Any for T where
    T: 'static + ?Sized, [src]

pub fn type_id(&self) -> TypeId[src]

Gets the TypeId of self. Read more

impl<T> Borrow<T> for T where
    T: ?Sized, [src]

pub fn borrow(&self) -> &T[src]

Immutably borrows from an owned value. Read more

impl<T> BorrowMut<T> for T where
    T: ?Sized, [src]

pub fn borrow_mut(&mut self) -> &mut T[src]

Mutably borrows from an owned value. Read more

impl<T> Downcast for T where
    T: Any, [src]

pub fn into_any(self: Box<T, Global>) -> Box<dyn Any + 'static, Global>ⓘ

Notable traits for Box<R, Global>

impl<R> Read for Box<R, Global> where
    R: Read + ?Sized, impl<W> Write for Box<W, Global> where
    W: Write + ?Sized, impl<F, A> Future for Box<F, A> where
    F: Future + Unpin + ?Sized,
    A: Allocator + 'static,  type Output = <F as Future>::Output;impl<I, A> Iterator for Box<I, A> where
    A: Allocator,
    I: Iterator + ?Sized,  type Item = <I as Iterator>::Item;
[src]

Convert Box<dyn Trait> (where Trait: Downcast) to Box<dyn Any>. Box<dyn Any> can then be further downcast into Box<ConcreteType> where ConcreteType implements Trait. Read more

pub fn into_any_rc(self: Rc<T>) -> Rc<dyn Any + 'static>[src]

Convert Rc<Trait> (where Trait: Downcast) to Rc<Any>. Rc<Any> can then be further downcast into Rc<ConcreteType> where ConcreteType implements Trait. Read more

pub fn as_any(&self) -> &(dyn Any + 'static)[src]

Convert &Trait (where Trait: Downcast) to &Any. This is needed since Rust cannot generate &Any’s vtable from &Trait’s. Read more

pub fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)[src]

Convert &mut Trait (where Trait: Downcast) to &Any. This is needed since Rust cannot generate &mut Any’s vtable from &mut Trait’s. Read more

impl<T> DowncastSync for T where
    T: Any + Send + Sync, [src]

pub fn into_any_arc(self: Arc<T>) -> Arc<dyn Any + 'static + Sync + Send>[src]

Convert Arc<Trait> (where Trait: Downcast) to Arc<Any>. Arc<Any> can then be further downcast into Arc<ConcreteType> where ConcreteType implements Trait. Read more

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

pub fn __clone_box(&self, Private) -> *mut ()[src]

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

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

Performs the conversion.

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

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

Performs the conversion.

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

type Owned = T

The resulting type after obtaining ownership.

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

Creates owned data from borrowed data, usually by cloning. Read more

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

🔬 This is a nightly-only experimental API. (toowned_clone_into)

recently added

Uses borrowed data to replace owned data, usually by cloning. Read more

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

type Error = Infallible

The type returned in the event of a conversion error.

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

Performs the conversion.

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

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.

pub fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>[src]

Performs the conversion.