Struct tract_hir::prelude::ModelPatch [−][src]
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, Global>
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, RandomState>
map of replaced inputs (patch node id to model node id)
incoming: HashMap<OutletId, OutletId, RandomState>
map of patch inputs to model wires
shunt_outlet_by: HashMap<OutletId, OutletId, RandomState>
map of old wires to be replaced by new wires
obliterate: Vec<usize, Global>
operations to discard from the model
Implementations
impl<F, O> ModelPatch<F, O> where
O: Display + Debug + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,
F: Fact + Clone + 'static + Hash,
Graph<F, O>: SpecialOps<F, O>,
[src]
O: Display + Debug + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,
F: Fact + Clone + 'static + Hash,
Graph<F, O>: SpecialOps<F, O>,
pub fn new(s: impl Into<String>) -> ModelPatch<F, O>
[src]
pub fn push_context(&mut self, s: impl Into<String>)
[src]
pub fn with_context(self, s: impl Into<String>) -> ModelPatch<F, O>
[src]
pub fn is_empty(&self) -> bool
[src]
pub fn tap_model(
&mut self,
model: &Graph<F, O>,
outlet: OutletId
) -> Result<OutletId, Error>
[src]
&mut self,
model: &Graph<F, O>,
outlet: OutletId
) -> Result<OutletId, Error>
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
) -> Result<(), Error>
[src]
&mut self,
outlet: OutletId,
by: OutletId
) -> Result<(), Error>
pub fn shunt_outside(
&mut self,
model: &Graph<F, O>,
outlet: OutletId,
by: OutletId
) -> Result<(), Error>
[src]
&mut self,
model: &Graph<F, O>,
outlet: OutletId,
by: OutletId
) -> Result<(), Error>
Replace an Outlet in the target model by one from the patch.
pub fn obliterate(&mut self, node: usize) -> Result<(), Error>
[src]
pub fn replace_single_op<IO>(
patched_model: &Graph<F, O>,
node: &Node<F, O>,
inputs: &[OutletId],
new_op: IO
) -> Result<ModelPatch<F, O>, Error> where
IO: Into<O>,
[src]
patched_model: &Graph<F, O>,
node: &Node<F, O>,
inputs: &[OutletId],
new_op: IO
) -> Result<ModelPatch<F, O>, Error> 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: &Node<F, O>,
new_op: IO
) -> Result<ModelPatch<F, O>, Error> where
IO: Into<O>,
[src]
patched_model: &Graph<F, O>,
node: &Node<F, O>,
new_op: IO
) -> Result<ModelPatch<F, O>, Error> 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: &Node<F, O>
) -> Result<ModelPatch<F, O>, Error>
[src]
patched_model: &Graph<F, O>,
node: &Node<F, O>
) -> Result<ModelPatch<F, O>, Error>
Convenience method creating a patch that shunt the given node.
pub fn single_unary_op<IO>(
patched_model: &Graph<F, O>,
node: &Node<F, O>,
new_op: IO
) -> Result<ModelPatch<F, O>, Error> where
IO: Into<O>,
[src]
patched_model: &Graph<F, O>,
node: &Node<F, O>,
new_op: IO
) -> Result<ModelPatch<F, O>, Error> 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>, Error> where
IO: Into<O>,
[src]
patched_model: &Graph<F, O>,
outlet: OutletId,
name: impl Into<String>,
new_op: IO,
fact: F
) -> Result<ModelPatch<F, O>, Error> 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<(), Error>
[src]
Apply all changes in the patch to the target model.
Methods from Deref<Target = Graph<F, O>>
pub fn add_node(
&mut self,
name: impl Into<String>,
op: impl Into<O>,
output_facts: SmallVec<[F; 4]>
) -> Result<usize, Error>
[src]
&mut self,
name: impl Into<String>,
op: impl Into<O>,
output_facts: SmallVec<[F; 4]>
) -> Result<usize, Error>
pub fn add_edge(
&mut self,
outlet: OutletId,
inlet: InletId
) -> Result<(), Error>
[src]
&mut self,
outlet: OutletId,
inlet: InletId
) -> Result<(), Error>
Connect a node outlet to a node inlet.
pub fn input_outlets(&self) -> Result<&[OutletId], Error>
[src]
Get model inputs.
pub fn set_input_outlets(&mut self, inputs: &[OutletId]) -> Result<(), Error>
[src]
Change model inputs.
pub fn set_input_names(
&mut self,
inputs: impl IntoIterator<Item = impl AsRef<str>>
) -> Result<(), Error>
[src]
&mut self,
inputs: impl IntoIterator<Item = impl AsRef<str>>
) -> Result<(), Error>
Set model inputs by the node name.
pub fn input_fact(&self, ix: usize) -> Result<&F, Error>
[src]
Get the ix
-th input tensor type information.
pub fn input_fact_mut(&mut self, ix: usize) -> Result<&mut F, Error>
[src]
Get the ix
-th input tensor type information, mutably.
pub fn set_input_fact(&mut self, input: usize, fact: F) -> Result<(), Error>
[src]
Set the ix
-th input tensor type information.
pub fn output_outlets(&self) -> Result<&[OutletId], Error>
[src]
Get model outputs.
pub fn auto_outputs(&mut self) -> Result<(), Error>
[src]
Guess outputs from the topology: node or nodes with no successors.
pub fn set_output_outlets(&mut self, outputs: &[OutletId]) -> Result<(), Error>
[src]
Change model outputs.
pub fn set_output_names(
&mut self,
outputs: impl IntoIterator<Item = impl AsRef<str>>
) -> Result<(), Error>
[src]
&mut self,
outputs: impl IntoIterator<Item = impl AsRef<str>>
) -> Result<(), Error>
Set model outputs by node names.
pub fn output_fact(&self, ix: usize) -> Result<&F, Error>
[src]
Get the ix
-th input tensor type information.
pub fn output_fact_mut(&mut self, ix: usize) -> Result<&mut F, Error>
[src]
Get the ix
-th input tensor type information, mutably.
pub fn set_output_fact(&mut self, output: usize, fact: F) -> Result<(), Error>
[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) -> Result<usize, Error>
[src]
pub fn node_by_name(&self, name: impl AsRef<str>) -> Result<&Node<F, O>, Error>
[src]
Find a node by its name.
pub fn node_by_name_mut(
&mut self,
name: impl AsRef<str>
) -> Result<&mut Node<F, O>, Error>
[src]
&mut self,
name: impl AsRef<str>
) -> Result<&mut Node<F, O>, Error>
Borrow mutably a node by its name.
pub fn rename_node(&mut self, id: usize, name: &str) -> Result<(), Error>
[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>]ⓘ
[src]
Access the nodes table.
pub fn nodes_mut(&mut self) -> &mut [Node<F, O>]ⓘ
[src]
Access the nodes table.
pub fn node_facts(
&self,
id: usize
) -> Result<(SmallVec<[&F; 4]>, SmallVec<[&F; 4]>), Error>
[src]
&self,
id: usize
) -> Result<(SmallVec<[&F; 4]>, SmallVec<[&F; 4]>), Error>
Get input and output tensor information for a node.
pub fn node_input_facts(
&self,
node_id: usize
) -> Result<SmallVec<[&F; 4]>, Error>
[src]
&self,
node_id: usize
) -> Result<SmallVec<[&F; 4]>, Error>
Get input tensor information for a node.
pub fn node_output_facts(
&self,
node_id: usize
) -> Result<SmallVec<[&F; 4]>, Error>
[src]
&self,
node_id: usize
) -> Result<SmallVec<[&F; 4]>, Error>
Get output tensor information for a node.
pub fn outlet_fact(&self, outlet: OutletId) -> Result<&F, Error>
[src]
Get tensor information for a single outlet.
pub fn outlet_fact_mut(&mut self, outlet: OutletId) -> Result<&mut F, Error>
[src]
Get tensor information for a single outlet.
pub fn outlets_fact_mut(
&mut self,
outlets: &[OutletId]
) -> Result<SmallVec<[&mut F; 4]>, Error>
[src]
&mut self,
outlets: &[OutletId]
) -> Result<SmallVec<[&mut F; 4]>, Error>
Get multiple mutable tensor information for outlets.
pub fn set_outlet_fact(
&mut self,
outlet: OutletId,
fact: F
) -> Result<(), Error>
[src]
&mut self,
outlet: OutletId,
fact: F
) -> Result<(), Error>
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
) -> Result<(), Error>
[src]
&mut self,
outlet: OutletId,
label: String
) -> Result<(), Error>
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) -> Result<Vec<usize, Global>, Error>
[src]
Computes an evalutation order for the graph inputs and outputs
pub fn check_edges(&self) -> Result<(), Error>
[src]
Performs a sanity check on network connections.
pub fn single_prec(&self, id: usize) -> Result<Option<&Node<F, O>>, Error>
[src]
pub fn single_prec_at(
&self,
id: usize,
count: usize
) -> Result<Option<&Node<F, O>>, Error>
[src]
&self,
id: usize,
count: usize
) -> Result<Option<&Node<F, O>>, Error>
pub fn single_succ_at(
&self,
id: usize,
count: usize
) -> Result<Option<&Node<F, O>>, Error>
[src]
&self,
id: usize,
count: usize
) -> Result<Option<&Node<F, O>>, Error>
pub fn single_succ(&self, id: usize) -> Result<Option<&Node<F, O>>, Error>
[src]
pub fn outlet_successors(&self, outlet: OutletId) -> &[InletId]ⓘ
[src]
pub fn add_const(
&mut self,
name: impl Into<String>,
v: impl IntoArcTensor
) -> Result<OutletId, Error>
[src]
&mut self,
name: impl Into<String>,
v: impl IntoArcTensor
) -> Result<OutletId, Error>
pub fn check_compact(&self) -> Result<(), Error>
[src]
pub fn compact(&self) -> Result<Graph<F, O>, Error>
[src]
pub fn signature(&self) -> u64
[src]
pub fn declutter(
&self
) -> Result<Graph<TypedFact, Box<dyn TypedOp + 'static, Global>>, Error>
[src]
&self
) -> Result<Graph<TypedFact, Box<dyn TypedOp + 'static, Global>>, Error>
Perform declutter passes on the network.
pub fn concretize_dims(
&self,
values: &SymbolValues
) -> Result<Graph<TypedFact, Box<dyn TypedOp + 'static, Global>>, Error>
[src]
&self,
values: &SymbolValues
) -> Result<Graph<TypedFact, Box<dyn TypedOp + 'static, Global>>, Error>
pub fn invariants(&self) -> Result<Invariants, Error>
[src]
Trait Implementations
impl<F, O> Clone for ModelPatch<F, O> where
O: Clone + Display + Debug + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + 'static + Hash,
F: Clone + Fact + 'static + Hash,
[src]
O: Clone + Display + Debug + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + 'static + Hash,
F: Clone + Fact + 'static + Hash,
pub fn clone(&self) -> ModelPatch<F, O>
[src]
pub fn clone_from(&mut self, source: &Self)
1.0.0[src]
impl<F, O> Debug for ModelPatch<F, O> where
O: Debug + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,
F: Debug + Fact + Clone + 'static + Hash,
[src]
O: Debug + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,
F: Debug + Fact + Clone + 'static + Hash,
impl<F, O> Default for ModelPatch<F, O> where
O: Display + Debug + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,
F: Fact + Clone + 'static + Hash,
[src]
O: Display + Debug + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,
F: Fact + Clone + 'static + Hash,
pub fn default() -> ModelPatch<F, O>
[src]
impl<F, O> Deref for ModelPatch<F, O> where
O: Display + Debug + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,
F: Fact + Clone + 'static + Hash,
[src]
O: Display + Debug + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,
F: Fact + Clone + 'static + Hash,
type Target = Graph<F, O>
The resulting type after dereferencing.
pub fn deref(&self) -> &Graph<F, O>
[src]
impl<F, O> DerefMut for ModelPatch<F, O> where
O: Display + Debug + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,
F: Fact + Clone + 'static + Hash,
[src]
O: Display + Debug + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,
F: Fact + Clone + 'static + Hash,
Auto Trait Implementations
impl<F, O> RefUnwindSafe for ModelPatch<F, O> where
F: RefUnwindSafe,
O: RefUnwindSafe,
F: RefUnwindSafe,
O: RefUnwindSafe,
impl<F, O> Send for ModelPatch<F, O> where
O: Send,
O: Send,
impl<F, O> Sync for ModelPatch<F, O> where
O: Sync,
O: Sync,
impl<F, O> Unpin for ModelPatch<F, O> where
F: Unpin,
O: Unpin,
F: Unpin,
O: Unpin,
impl<F, O> UnwindSafe for ModelPatch<F, O> where
F: RefUnwindSafe + UnwindSafe,
O: UnwindSafe,
F: RefUnwindSafe + UnwindSafe,
O: UnwindSafe,
Blanket Implementations
impl<T> Any for T where
T: 'static + ?Sized,
[src]
T: 'static + ?Sized,
impl<T> Borrow<T> for T where
T: ?Sized,
[src]
T: ?Sized,
pub fn borrow(&self) -> &TⓘNotable traits for &'_ mut R
impl<'_, R> Read for &'_ mut R where
R: Read + ?Sized, impl<'_, W> Write for &'_ mut W where
W: Write + ?Sized, impl<'_, F> Future for &'_ mut F where
F: Future + Unpin + ?Sized, type Output = <F as Future>::Output;impl<'_, I> Iterator for &'_ mut I where
I: Iterator + ?Sized, type Item = <I as Iterator>::Item;
[src]
Notable traits for &'_ mut R
impl<'_, R> Read for &'_ mut R where
R: Read + ?Sized, impl<'_, W> Write for &'_ mut W where
W: Write + ?Sized, impl<'_, F> Future for &'_ mut F where
F: Future + Unpin + ?Sized, type Output = <F as Future>::Output;impl<'_, I> Iterator for &'_ mut I where
I: Iterator + ?Sized, type Item = <I as Iterator>::Item;
impl<T> BorrowMut<T> for T where
T: ?Sized,
[src]
T: ?Sized,
pub fn borrow_mut(&mut self) -> &mut TⓘNotable traits for &'_ mut R
impl<'_, R> Read for &'_ mut R where
R: Read + ?Sized, impl<'_, W> Write for &'_ mut W where
W: Write + ?Sized, impl<'_, F> Future for &'_ mut F where
F: Future + Unpin + ?Sized, type Output = <F as Future>::Output;impl<'_, I> Iterator for &'_ mut I where
I: Iterator + ?Sized, type Item = <I as Iterator>::Item;
[src]
Notable traits for &'_ mut R
impl<'_, R> Read for &'_ mut R where
R: Read + ?Sized, impl<'_, W> Write for &'_ mut W where
W: Write + ?Sized, impl<'_, F> Future for &'_ mut F where
F: Future + Unpin + ?Sized, type Output = <F as Future>::Output;impl<'_, I> Iterator for &'_ mut I where
I: Iterator + ?Sized, type Item = <I as Iterator>::Item;
impl<T> Downcast for T where
T: Any,
T: Any,
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
I: Iterator + ?Sized,
A: Allocator, type Item = <I as Iterator>::Item;
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
I: Iterator + ?Sized,
A: Allocator, type Item = <I as Iterator>::Item;
pub fn into_any_rc(self: Rc<T>) -> Rc<dyn Any + 'static>
pub fn as_any(&self) -> &(dyn Any + 'static)
pub fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)
impl<T> DowncastSync for T where
T: Any + Send + Sync,
T: Any + Send + Sync,
impl<T> DynClone for T where
T: Clone,
[src]
T: Clone,
pub fn __clone_box(&self, Private) -> *mut ()
[src]
impl<T> From<T> for T
[src]
impl<T, U> Into<U> for T where
U: From<T>,
[src]
U: From<T>,
impl<T> ToOwned for T where
T: Clone,
[src]
T: Clone,
type Owned = T
The resulting type after obtaining ownership.
pub fn to_owned(&self) -> T
[src]
pub fn clone_into(&self, target: &mut T)
[src]
impl<T, U> TryFrom<U> for T where
U: Into<T>,
[src]
U: Into<T>,
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]
impl<T, U> TryInto<U> for T where
U: TryFrom<T>,
[src]
U: TryFrom<T>,