Struct tract_hir::prelude::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>,
    // some fields omitted
}

Fields

states: Vec<Option<Box<dyn OpState + 'static, Global>>, Global>

session_state: SessionState

values: 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, 

pub fn new(plan: P) -> Result<SimpleState<F, O, M, P>, Error>

pub fn reset_wires(&mut self) -> Result<(), Error>

Reset wires state.

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 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 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>>, 

pub fn clone(&self) -> SimpleState<F, O, M, P>

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<F, O, M, P> Debug for SimpleState<F, O, M, P> where
    F: Debug + 'static + Fact + Hash + Clone,
    O: Debug + 'static + Display + AsRef<dyn Op + 'static> + AsMut<dyn Op + 'static> + Clone + Hash,
    M: Debug + Borrow<Graph<F, O>> + Hash,
    P: Debug + Borrow<SimplePlan<F, O, M>>, 

pub fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter.