Struct tract_pulse::internal::SimpleState
[−]pub struct SimpleState<F, O, M, P> where
F: 'static + Fact + Hash + Clone,
O: 'static + Debug + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + Hash,
M: Borrow<Graph<F, O>> + Hash,
P: Borrow<SimplePlan<F, O, M>>, {
pub states: Vec<Option<Box<dyn OpState + 'static, Global>>, Global>,
pub session_state: SessionState,
pub values: Vec<Option<SmallVec<[Arc<Tensor>; 4]>>, Global>,
/* private fields */
}Fields
states: Vec<Option<Box<dyn OpState + 'static, Global>>, Global>session_state: SessionStatevalues: Vec<Option<SmallVec<[Arc<Tensor>; 4]>>, Global>Implementations
impl<F, O, M, P> SimpleState<F, O, M, P> where
F: 'static + Fact + Hash + Clone,
O: 'static + Debug + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + 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: 'static + Fact + Hash + Clone,
O: 'static + Debug + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + Hash,
M: Borrow<Graph<F, O>> + Hash,
P: Borrow<SimplePlan<F, O, M>> + Clone,
pub fn new(plan: P) -> Result<SimpleState<F, O, M, P>, Error>
pub fn reset_wires(&mut self) -> Result<(), Error>
pub fn reset_wires(&mut self) -> Result<(), Error>
Reset wires state.
pub fn reset_op_states(&mut self) -> Result<(), Error>
pub fn reset_op_states(&mut self) -> Result<(), Error>
Reset wires state.
pub fn run(
&mut self,
inputs: SmallVec<[Tensor; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, Error>
pub fn exec(&mut self) -> Result<SmallVec<[Arc<Tensor>; 4]>, Error>
pub fn run_plan_with_eval<Eval, E>(
&mut self,
inputs: SmallVec<[Tensor; 4]>,
eval: Eval
) -> Result<SmallVec<[Arc<Tensor>; 4]>, Error> where
Eval: for<'a, 'b, 'c> FnMut(&'a mut SessionState, Option<&'b mut (dyn OpState + 'static)>, &'c Node<F, O>, SmallVec<[Arc<Tensor>; 4]>) -> Result<SmallVec<[Arc<Tensor>; 4]>, E>,
E: 'static + Into<Error> + Send + Sync,
pub fn exec_plan_with_eval<Eval, E>(
&mut self,
eval: Eval
) -> Result<SmallVec<[Arc<Tensor>; 4]>, Error> where
Eval: for<'c, 'a, 'b> FnMut(&'a mut SessionState, Option<&'b mut (dyn OpState + 'static)>, &'c Node<F, O>, SmallVec<[Arc<Tensor>; 4]>) -> Result<SmallVec<[Arc<Tensor>; 4]>, E>,
E: 'static + Into<Error> + Send + Sync,
pub fn set_inputs(&mut self, inputs: SmallVec<[Tensor; 4]>) -> Result<(), Error>
pub fn set_input(&mut self, input: usize, t: Tensor) -> Result<(), Error>
pub fn take_outputs(&mut self) -> Result<Vec<Arc<Tensor>, Global>, Error>
pub fn set_values(
&mut self,
id: usize,
values: SmallVec<[Tensor; 4]>
) -> Result<(), Error>
pub fn set_value(&mut self, id: usize, value: Tensor) -> Result<(), Error>
pub fn prepare_inputs(
&self,
node: usize
) -> Result<SmallVec<[Arc<Tensor>; 4]>, Error>
pub fn compute_one(&mut self, node: usize) -> Result<(), Error>
pub fn compute_one_with_inputs(
&mut self,
node: usize,
inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<(), Error>
pub fn compute_recursively(
&mut self,
node: usize
) -> Result<&[Arc<Tensor>], Error>
pub fn take_by_name(
&mut self,
name: &str
) -> Result<SmallVec<[Tensor; 4]>, Error>
pub fn take(&mut self, id: usize) -> Result<SmallVec<[Tensor; 4]>, Error>
pub fn plan(&self) -> &SimplePlan<F, O, M>
pub fn model(&self) -> &Graph<F, O>
Trait Implementations
impl<F, O, M, P> Clone for SimpleState<F, O, M, P> where
F: Clone + 'static + Fact + Hash,
O: Clone + 'static + Debug + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Hash,
M: Clone + Borrow<Graph<F, O>> + Hash,
P: Clone + Borrow<SimplePlan<F, O, M>>,
impl<F, O, M, P> Clone for SimpleState<F, O, M, P> where
F: Clone + 'static + Fact + Hash,
O: Clone + 'static + Debug + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Hash,
M: Clone + Borrow<Graph<F, O>> + Hash,
P: Clone + Borrow<SimplePlan<F, O, M>>,
fn 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<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<I, A> Iterator for Box<I, A> where
I: Iterator + ?Sized,
A: Allocator, type Item = <I as Iterator>::Item;impl<F, A> Future for Box<F, A> where
F: Future + Unpin + ?Sized,
A: Allocator + 'static, type Output = <F as Future>::Output;
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<I, A> Iterator for Box<I, A> where
I: Iterator + ?Sized,
A: Allocator, type Item = <I as Iterator>::Item;impl<F, A> Future for Box<F, A> where
F: Future + Unpin + ?Sized,
A: Allocator + 'static, type Output = <F as Future>::Output;
R: Read + ?Sized, impl<W> Write for Box<W, Global> where
W: Write + ?Sized, impl<I, A> Iterator for Box<I, A> where
I: Iterator + ?Sized,
A: Allocator, type Item = <I as Iterator>::Item;impl<F, A> Future for Box<F, A> where
F: Future + Unpin + ?Sized,
A: Allocator + 'static, type Output = <F as Future>::Output;
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