Struct tract_core::plan::SimpleState
source · [−]pub struct SimpleState<F, O, M, P> where
F: Fact + Hash + Clone + 'static,
O: Debug + Display + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static + Hash,
M: Borrow<Graph<F, O>> + Hash,
P: Borrow<SimplePlan<F, O, M>>, {
pub states: Vec<Option<Box<dyn OpState>>>,
pub session_state: SessionState,
pub values: Vec<Option<TVec<Arc<Tensor>>>>,
/* private fields */
}Fields
states: Vec<Option<Box<dyn OpState>>>session_state: SessionStatevalues: Vec<Option<TVec<Arc<Tensor>>>>Implementations
sourceimpl<F, O, M, P> SimpleState<F, O, M, P> where
F: Fact + Hash + Clone + 'static,
O: Debug + Display + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static + Hash,
M: Borrow<Graph<F, O>> + Hash,
P: Borrow<SimplePlan<F, O, M>> + Clone,
impl<F, O, M, P> SimpleState<F, O, M, P> where
F: Fact + Hash + Clone + 'static,
O: Debug + Display + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static + Hash,
M: Borrow<Graph<F, O>> + Hash,
P: Borrow<SimplePlan<F, O, M>> + Clone,
pub fn new(plan: P) -> TractResult<SimpleState<F, O, M, P>>
sourcepub fn reset_wires(&mut self) -> TractResult<()>
pub fn reset_wires(&mut self) -> TractResult<()>
Reset wires state.
sourcepub fn reset_op_states(&mut self) -> TractResult<()>
pub fn reset_op_states(&mut self) -> TractResult<()>
Reset wires state.
pub fn run(&mut self, inputs: TVec<Tensor>) -> TractResult<TVec<Arc<Tensor>>>
pub fn run_plan_with_eval<Eval, E>(
&mut self,
inputs: TVec<Tensor>,
eval: Eval
) -> TractResult<TVec<Arc<Tensor>>> where
Eval: for<'a, 'b, 'c> FnMut(&'a mut SessionState, Option<&'b mut (dyn OpState + 'static)>, &'c Node<F, O>, TVec<Arc<Tensor>>) -> Result<TVec<Arc<Tensor>>, E>,
E: Into<Error> + Send + Sync + 'static,
pub fn set_inputs(&mut self, inputs: TVec<Tensor>) -> TractResult<()>
pub fn set_input(&mut self, input: usize, t: Tensor) -> TractResult<()>
pub fn take_outputs(&mut self) -> TractResult<Vec<Arc<Tensor>>>
pub fn set_values(&mut self, id: usize, values: TVec<Tensor>) -> TractResult<()>
pub fn set_value(&mut self, id: usize, value: Tensor) -> TractResult<()>
pub fn prepare_inputs(&self, node: usize) -> TractResult<TVec<Arc<Tensor>>>
pub fn compute_one(&mut self, node: usize) -> TractResult<()>
pub fn compute_one_with_inputs(
&mut self,
node: usize,
inputs: TVec<Arc<Tensor>>
) -> TractResult<()>
pub fn compute_recursively(&mut self, node: usize) -> TractResult<&[Arc<Tensor>]>
pub fn take_by_name(&mut self, name: &str) -> TractResult<TVec<Tensor>>
pub fn take(&mut self, id: usize) -> TractResult<TVec<Tensor>>
pub fn plan(&self) -> &SimplePlan<F, O, M>
pub fn model(&self) -> &Graph<F, O>
Trait Implementations
sourceimpl<F: Clone, O: Clone, M: Clone, P: Clone> Clone for SimpleState<F, O, M, P> where
F: Fact + Hash + Clone + 'static,
O: Debug + Display + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static + Hash,
M: Borrow<Graph<F, O>> + Hash,
P: Borrow<SimplePlan<F, O, M>>,
impl<F: Clone, O: Clone, M: Clone, P: Clone> Clone for SimpleState<F, O, M, P> where
F: Fact + Hash + Clone + 'static,
O: Debug + Display + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static + Hash,
M: Borrow<Graph<F, O>> + Hash,
P: Borrow<SimplePlan<F, O, M>>,
sourcefn clone(&self) -> SimpleState<F, O, M, P>
fn clone(&self) -> SimpleState<F, O, M, P>
Returns a copy of the value. Read more
1.0.0 · sourcefn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from source. Read more
Auto Trait Implementations
impl<F, O, M, P> !RefUnwindSafe for SimpleState<F, O, M, P>
impl<F, O, M, P> Send for SimpleState<F, O, M, P> where
M: Send,
O: Send,
P: Send,
impl<F, O, M, P> !Sync for SimpleState<F, O, M, P>
impl<F, O, M, P> Unpin for SimpleState<F, O, M, P> where
F: Unpin,
M: Unpin,
O: Unpin,
P: Unpin,
impl<F, O, M, P> !UnwindSafe for SimpleState<F, O, M, P>
Blanket Implementations
sourceimpl<T> BorrowMut<T> for T where
T: ?Sized,
impl<T> BorrowMut<T> for T where
T: ?Sized,
const: unstable · sourcefn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
sourceimpl<T> Downcast for T where
T: Any,
impl<T> Downcast for T where
T: Any,
sourcefn into_any(self: Box<T, Global>) -> Box<dyn Any + 'static, Global>ⓘNotable traits for Box<W, Global>impl<W> Write for Box<W, Global> where
W: Write + ?Sized, impl<R> Read for Box<R, Global> where
R: Read + ?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;
fn into_any(self: Box<T, Global>) -> Box<dyn Any + 'static, Global>ⓘNotable traits for Box<W, Global>impl<W> Write for Box<W, Global> where
W: Write + ?Sized, impl<R> Read for Box<R, Global> where
R: Read + ?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;
W: Write + ?Sized, impl<R> Read for Box<R, Global> where
R: Read + ?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;
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
sourcefn into_any_rc(self: Rc<T>) -> Rc<dyn Any + 'static>
fn into_any_rc(self: Rc<T>) -> Rc<dyn Any + 'static>
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
sourcefn as_any(&self) -> &(dyn Any + 'static)
fn as_any(&self) -> &(dyn Any + 'static)
Convert &Trait (where Trait: Downcast) to &Any. This is needed since Rust cannot
generate &Any’s vtable from &Trait’s. Read more
sourcefn as_any_mut(&mut self) -> &mut (dyn Any + 'static)
fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)
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