[−][src]Struct tract_hir::internal::dsl::ModelImpl
Main model class
Parameterized by a Fact class.
Fields
label: Option<String>
nodes: Vec<BaseNode<F, O>>
all nodes in the model
inputs: Vec<OutletId>
model inputs
outputs: Vec<OutletId>
model outputs
outlet_labels: HashMap<OutletId, String, RandomState>
outlet labels
Implementations
impl<F, O> ModelImpl<F, O> where
F: Fact + Hash + Clone + 'static,
O: Debug + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,
ModelImpl<F, O>: Model,
[src]
F: Fact + Hash + Clone + 'static,
O: Debug + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,
ModelImpl<F, O>: Model,
pub fn add_node(
&mut self,
name: impl Into<String>,
op: impl Into<O>,
output_facts: SmallVec<[F; 4]>
) -> Result<usize, TractError>
[src]
&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>
[src]
&mut self,
outlet: OutletId,
inlet: InletId
) -> Result<(), TractError>
Connect a node outlet to a node inlet.
pub fn input_outlets(&self) -> Result<&[OutletId], TractError>
[src]
Get model inputs.
pub fn set_input_outlets(
&mut self,
inputs: &[OutletId]
) -> Result<(), TractError>
[src]
&mut self,
inputs: &[OutletId]
) -> Result<(), TractError>
Change model inputs.
pub fn with_input_outlets(
self,
inputs: &[OutletId]
) -> Result<ModelImpl<F, O>, TractError>
[src]
self,
inputs: &[OutletId]
) -> Result<ModelImpl<F, O>, TractError>
Change model inputs and return self
.
pub fn set_input_names(
&mut self,
inputs: impl IntoIterator<Item = impl AsRef<str>>
) -> Result<(), TractError>
[src]
&mut self,
inputs: impl IntoIterator<Item = impl AsRef<str>>
) -> Result<(), TractError>
Set model inputs by the node name.
pub fn with_input_names(
self,
inputs: impl IntoIterator<Item = impl AsRef<str>>
) -> Result<ModelImpl<F, O>, TractError>
[src]
self,
inputs: impl IntoIterator<Item = impl AsRef<str>>
) -> Result<ModelImpl<F, O>, TractError>
Set model inputs by the node name and return self
.
pub fn input_fact(&self, ix: usize) -> Result<&F, TractError>
[src]
Get the ix
-th input tensor type information.
pub fn input_fact_mut(&mut self, ix: usize) -> Result<&mut F, TractError>
[src]
Get the ix
-th input tensor type information, mutably.
pub fn set_input_fact(
&mut self,
input: usize,
fact: F
) -> Result<(), TractError>
[src]
&mut self,
input: usize,
fact: F
) -> Result<(), TractError>
Set the ix
-th input tensor type information.
pub fn with_input_fact(
self,
input: usize,
fact: F
) -> Result<ModelImpl<F, O>, TractError>
[src]
self,
input: usize,
fact: F
) -> Result<ModelImpl<F, O>, TractError>
Set the ix
-th input tensor type information and return self
.
pub fn output_outlets(&self) -> Result<&[OutletId], TractError>
[src]
Get model outputs.
pub fn auto_outputs(&mut self) -> Result<(), TractError>
[src]
Guess outputs from the topology: node or nodes with no successors.
pub fn set_output_outlets(
&mut self,
outputs: &[OutletId]
) -> Result<(), TractError>
[src]
&mut self,
outputs: &[OutletId]
) -> Result<(), TractError>
Change model outputs.
pub fn with_output_outlets(
self,
outputs: &[OutletId]
) -> Result<ModelImpl<F, O>, TractError>
[src]
self,
outputs: &[OutletId]
) -> Result<ModelImpl<F, O>, TractError>
Change model outputs and return self
.
pub fn set_output_names(
&mut self,
outputs: impl IntoIterator<Item = impl AsRef<str>>
) -> Result<(), TractError>
[src]
&mut self,
outputs: impl IntoIterator<Item = impl AsRef<str>>
) -> Result<(), TractError>
Set model outputs by node names.
pub fn with_output_names(
self,
outputs: impl IntoIterator<Item = impl AsRef<str>>
) -> Result<ModelImpl<F, O>, TractError>
[src]
self,
outputs: impl IntoIterator<Item = impl AsRef<str>>
) -> Result<ModelImpl<F, O>, TractError>
Set model outputs by node names and return self
.
pub fn output_fact(&self, ix: usize) -> Result<&F, TractError>
[src]
Get the ix
-th input tensor type information.
pub fn output_fact_mut(&mut self, ix: usize) -> Result<&mut F, TractError>
[src]
Get the ix
-th input tensor type information, mutably.
pub fn set_output_fact(
&mut self,
output: usize,
fact: F
) -> Result<(), TractError>
[src]
&mut self,
output: usize,
fact: F
) -> Result<(), TractError>
Set the ix
-th output tensor type information.
pub fn with_output_fact(
self,
output: usize,
fact: F
) -> Result<ModelImpl<F, O>, TractError>
[src]
self,
output: usize,
fact: F
) -> Result<ModelImpl<F, O>, TractError>
Set the ix
-th output tensor type information and return self
.
pub fn node_names(&self) -> impl Iterator<Item = &str>
[src]
Iterate over all node names.
pub fn node_by_name<S>(&self, name: S) -> Result<&BaseNode<F, O>, TractError> where
S: AsRef<str>,
[src]
S: AsRef<str>,
Find a node by its name.
pub fn node_by_name_mut(
&mut self,
name: &str
) -> Result<&mut BaseNode<F, O>, TractError>
[src]
&mut self,
name: &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>
[src]
pub fn node(&self, id: usize) -> &BaseNode<F, O>
[src]
Find a node by its id.
pub fn node_mut(&mut self, id: usize) -> &mut BaseNode<F, O>
[src]
Find a node by its id.
pub fn nodes(&self) -> &[BaseNode<F, O>]
[src]
Access the nodes table.
pub fn nodes_mut(&mut self) -> &mut [BaseNode<F, O>]
[src]
Access the nodes table.
pub fn node_facts(
&self,
id: usize
) -> Result<(SmallVec<[&F; 4]>, SmallVec<[&F; 4]>), TractError>
[src]
&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>
[src]
&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>
[src]
&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>
[src]
Get tensor information for a single outlet.
pub fn outlet_fact_mut(
&mut self,
outlet: OutletId
) -> Result<&mut F, TractError>
[src]
&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>
[src]
&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>
[src]
&mut self,
outlet: OutletId,
fact: F
) -> Result<(), TractError>
Set tensor information for a single outlet.
pub fn with_outlet_fact(
self,
outlet: OutletId,
fact: F
) -> Result<ModelImpl<F, O>, TractError>
[src]
self,
outlet: OutletId,
fact: F
) -> Result<ModelImpl<F, O>, TractError>
Set tensor information for a single outlet and return self
.
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<(), TractError>
[src]
&mut self,
outlet: OutletId,
label: String
) -> Result<(), TractError>
Set label for an outlet.
pub fn with_outlet_label(
self,
outlet: OutletId,
label: String
) -> Result<ModelImpl<F, O>, TractError>
[src]
self,
outlet: OutletId,
label: String
) -> Result<ModelImpl<F, O>, TractError>
Set label for an outlet and return self
.
pub fn find_outlet_label(&self, label: &str) -> Option<OutletId>
[src]
Find outlet by label.
pub fn eval_order(&self) -> Result<Vec<usize>, TractError>
[src]
Computes an evalutation order for the graph inputs and outputs
pub fn check_edges(&self) -> Result<(), TractError>
[src]
Performs a sanity check on network connections.
pub fn into_runnable(
self
) -> Result<SimplePlan<F, O, ModelImpl<F, O>>, TractError>
[src]
self
) -> Result<SimplePlan<F, O, ModelImpl<F, O>>, TractError>
Converts the model into a RunnableModel
which fixes the inputs and outputs and allows passing data through the model.
impl ModelImpl<TypedFact, Box<dyn TypedOp + 'static>>
[src]
pub fn signature(&self) -> u64
[src]
pub fn declutter(
self
) -> Result<ModelImpl<TypedFact, Box<dyn TypedOp + 'static>>, TractError>
[src]
self
) -> Result<ModelImpl<TypedFact, Box<dyn TypedOp + 'static>>, TractError>
Perform declutter pass on the network.
pub fn codegen(
self
) -> Result<ModelImpl<TypedFact, Box<dyn TypedOp + 'static>>, TractError>
[src]
self
) -> Result<ModelImpl<TypedFact, Box<dyn TypedOp + 'static>>, TractError>
Translate the graph to optimized operators.
pub fn invariants(&self) -> Result<Invariants, TractError>
[src]
pub fn into_normalized(
self
) -> Result<ModelImpl<NormalizedFact, Box<dyn TypedOp + 'static>>, TractError>
[src]
self
) -> Result<ModelImpl<NormalizedFact, Box<dyn TypedOp + 'static>>, TractError>
Attempt to convert the network to a NormalizedModel.
pub fn into_optimized(
self
) -> Result<ModelImpl<TypedFact, Box<dyn TypedOp + 'static>>, TractError>
[src]
self
) -> Result<ModelImpl<TypedFact, Box<dyn TypedOp + 'static>>, TractError>
Declutter as much as possible, then translate to optimized operators.
impl ModelImpl<NormalizedFact, Box<dyn TypedOp + 'static>>
[src]
pub fn into_typed(
self
) -> Result<ModelImpl<TypedFact, Box<dyn TypedOp + 'static>>, TractError>
[src]
self
) -> Result<ModelImpl<TypedFact, Box<dyn TypedOp + 'static>>, TractError>
Convert back to TypedModel.
Can not fail.
impl ModelImpl<PulsedFact, Box<dyn PulsedOp + 'static>>
[src]
pub fn new(
source: &ModelImpl<NormalizedFact, Box<dyn TypedOp + 'static>>,
pulse: usize
) -> Result<ModelImpl<PulsedFact, Box<dyn PulsedOp + 'static>>, TractError>
[src]
source: &ModelImpl<NormalizedFact, Box<dyn TypedOp + 'static>>,
pulse: usize
) -> Result<ModelImpl<PulsedFact, Box<dyn PulsedOp + 'static>>, TractError>
pub fn new_with_mapping(
source: &ModelImpl<NormalizedFact, Box<dyn TypedOp + 'static>>,
pulse: usize
) -> Result<(ModelImpl<PulsedFact, Box<dyn PulsedOp + 'static>>, HashMap<OutletId, OutletId, RandomState>), TractError>
[src]
source: &ModelImpl<NormalizedFact, Box<dyn TypedOp + 'static>>,
pulse: usize
) -> Result<(ModelImpl<PulsedFact, Box<dyn PulsedOp + 'static>>, HashMap<OutletId, OutletId, RandomState>), TractError>
pub fn into_typed(
self
) -> Result<ModelImpl<TypedFact, Box<dyn TypedOp + 'static>>, TractError>
[src]
self
) -> Result<ModelImpl<TypedFact, Box<dyn TypedOp + 'static>>, TractError>
Trait Implementations
impl<F, O> Clone for ModelImpl<F, O> where
F: Clone + Fact + Hash + 'static,
O: Clone + Debug + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + 'static + Hash,
[src]
F: Clone + Fact + Hash + 'static,
O: Clone + Debug + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + 'static + Hash,
impl<F, O> Debug for ModelImpl<F, O> where
F: Debug + Fact + Hash + Clone + 'static,
O: Debug + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,
[src]
F: Debug + Fact + Hash + Clone + 'static,
O: Debug + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,
impl<F, O> Default for ModelImpl<F, O> where
F: Fact + Hash + Clone + 'static,
O: Debug + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,
[src]
F: Fact + Hash + Clone + 'static,
O: Debug + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,
impl<F, O> Display for ModelImpl<F, O> where
F: Fact + Hash + Clone + 'static,
O: Debug + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,
[src]
F: Fact + Hash + Clone + 'static,
O: Debug + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,
impl<F, O> DynHash for ModelImpl<F, O> where
F: Fact + Hash + Clone + 'static,
O: Debug + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,
[src]
F: Fact + Hash + Clone + 'static,
O: Debug + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,
impl<F, O> Hash for ModelImpl<F, O> where
F: Fact + Hash + Clone + 'static,
O: Debug + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,
[src]
F: Fact + Hash + Clone + 'static,
O: Debug + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,
fn hash<H>(&self, state: &mut H) where
H: Hasher,
[src]
H: Hasher,
fn hash_slice<H>(data: &[Self], state: &mut H) where
H: Hasher,
1.3.0[src]
H: Hasher,
impl<F, O> Model for ModelImpl<F, O> where
F: Fact + Hash + Clone + 'static,
O: Debug + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,
[src]
F: Fact + Hash + Clone + 'static,
O: Debug + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,
fn model_label(&self) -> Option<&str>
[src]
fn node_id_by_name(&self, name: &str) -> Result<usize, TractError>
[src]
fn node_name(&self, id: usize) -> &str
[src]
fn node_inputs(&self, id: usize) -> &[OutletId]
[src]
fn node_output_count(&self, id: usize) -> usize
[src]
fn nodes_len(&self) -> usize
[src]
fn node_display(&self, id: usize) -> String
[src]
fn node_debug(&self, id: usize) -> String
[src]
fn eval_order(&self) -> Result<Vec<usize>, TractError>
[src]
fn eval_order_for_io(
&self,
inputs: &[usize],
outputs: &[usize]
) -> Result<Vec<usize>, TractError>
[src]
&self,
inputs: &[usize],
outputs: &[usize]
) -> Result<Vec<usize>, TractError>
fn input_outlets(&self) -> &[OutletId]
[src]
fn output_outlets(&self) -> &[OutletId]
[src]
fn node_op(&self, id: usize) -> &(dyn Op + 'static)
[src]
fn outlet_typedfact(&self, outlet: OutletId) -> Result<TypedFact, TractError>
[src]
fn outlet_fact_format(&self, outlet: OutletId) -> String
[src]
fn outlet_label(&self, id: OutletId) -> Option<&str>
[src]
fn outlet_successors(&self, outlet: OutletId) -> &[InletId]
[src]
fn nested_models(
&self,
node: usize
) -> Vec<(Cow<str>, &(dyn Model + 'static), Vec<String>, Vec<String>)>
[src]
&self,
node: usize
) -> Vec<(Cow<str>, &(dyn Model + 'static), Vec<String>, Vec<String>)>
impl<F, O> ModelDsl<F, O> for ModelImpl<F, O> where
F: Fact + Clone + 'static + Hash,
O: Debug + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,
[src]
F: Fact + Clone + 'static + Hash,
O: Debug + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 'static + Hash,
fn single_prec(&self, id: usize) -> Result<Option<&BaseNode<F, O>>, TractError>
[src]
fn single_prec_at(
&self,
id: usize,
count: usize
) -> Result<Option<&BaseNode<F, O>>, TractError>
[src]
&self,
id: usize,
count: usize
) -> Result<Option<&BaseNode<F, O>>, TractError>
fn single_succ_at(
&self,
id: usize,
count: usize
) -> Result<Option<&BaseNode<F, O>>, TractError>
[src]
&self,
id: usize,
count: usize
) -> Result<Option<&BaseNode<F, O>>, TractError>
fn single_succ(&self, id: usize) -> Result<Option<&BaseNode<F, O>>, TractError>
[src]
impl<F, O> ModelDslConst for ModelImpl<F, O> where
F: 'static + Fact + Clone + From<Arc<Tensor>> + Hash,
O: Debug + Display + From<Const> + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + Hash + 'static,
[src]
F: 'static + Fact + Clone + From<Arc<Tensor>> + Hash,
O: Debug + Display + From<Const> + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + Hash + 'static,
fn add_const(
&mut self,
name: impl Into<String>,
v: impl IntoArcTensor
) -> Result<OutletId, TractError>
[src]
&mut self,
name: impl Into<String>,
v: impl IntoArcTensor
) -> Result<OutletId, TractError>
impl ModelSpecialOps<PulsedFact, Box<dyn TypedOp + 'static>> for ModelImpl<PulsedFact, Box<dyn PulsedOp + 'static>>
[src]
fn add_source(
&mut self,
name: impl Into<String>,
fact: PulsedFact
) -> Result<OutletId, TractError>
[src]
&mut self,
name: impl Into<String>,
fact: PulsedFact
) -> Result<OutletId, TractError>
fn is_source(op: &(dyn Op + 'static)) -> bool
[src]
fn create_dummy(&self) -> Box<dyn TypedOp + 'static>
[src]
impl ModelSpecialOps<TypedFact, Box<dyn TypedOp + 'static>> for ModelImpl<TypedFact, Box<dyn TypedOp + 'static>>
[src]
fn add_source(
&mut self,
name: impl Into<String>,
fact: TypedFact
) -> Result<OutletId, TractError>
[src]
&mut self,
name: impl Into<String>,
fact: TypedFact
) -> Result<OutletId, TractError>
fn is_source(op: &(dyn Op + 'static)) -> bool
[src]
fn create_dummy(&self) -> Box<dyn TypedOp + 'static>
[src]
impl ModelWireNode<PulsedFact, Box<dyn PulsedOp + 'static>> for ModelImpl<PulsedFact, Box<dyn PulsedOp + 'static>>
[src]
fn wire_node(
&mut self,
name: impl Into<String>,
op: impl Into<Box<dyn PulsedOp + 'static>>,
inputs: &[OutletId]
) -> Result<SmallVec<[OutletId; 4]>, TractError>
[src]
&mut self,
name: impl Into<String>,
op: impl Into<Box<dyn PulsedOp + 'static>>,
inputs: &[OutletId]
) -> Result<SmallVec<[OutletId; 4]>, TractError>
impl ModelWireNode<TypedFact, Box<dyn TypedOp + 'static>> for ModelImpl<TypedFact, Box<dyn TypedOp + 'static>>
[src]
Auto Trait Implementations
impl<F, O> RefUnwindSafe for ModelImpl<F, O> where
F: RefUnwindSafe,
O: RefUnwindSafe,
F: RefUnwindSafe,
O: RefUnwindSafe,
impl<F, O> Send for ModelImpl<F, O> where
O: Send,
O: Send,
impl<F, O> Sync for ModelImpl<F, O> where
O: Sync,
O: Sync,
impl<F, O> Unpin for ModelImpl<F, O> where
F: Unpin,
O: Unpin,
F: Unpin,
O: Unpin,
impl<F, O> UnwindSafe for ModelImpl<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,
impl<T> BorrowMut<T> for T where
T: ?Sized,
[src]
T: ?Sized,
fn borrow_mut(&mut self) -> &mut T
[src]
impl<T> Downcast for T where
T: Any,
T: Any,
fn into_any(self: Box<T>) -> Box<dyn Any + 'static>
fn into_any_rc(self: Rc<T>) -> Rc<dyn Any + 'static>
fn as_any(&self) -> &(dyn Any + 'static)
fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)
impl<T> DowncastSync for T where
T: Send + Sync + Any,
T: Send + Sync + Any,
impl<T> DynClone for T where
T: Clone,
[src]
T: Clone,
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.
fn to_owned(&self) -> T
[src]
fn clone_into(&self, target: &mut T)
[src]
impl<T> ToString for T where
T: Display + ?Sized,
[src]
T: Display + ?Sized,
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.
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>,