Struct tract_core::plan::SimpleState

pub struct SimpleState<M: Borrow<Model>, P: Borrow<SimplePlan<M>>> {
    pub states: Vec<Option<Box<dyn OpState>>>,
    pub values: Vec<Option<TVec<SharedTensor>>>,
    // some fields omitted
}

Fields

states: Vec<Option<Box<dyn OpState>>>

values: Vec<Option<TVec<SharedTensor>>>

Methods

impl<M: Borrow<Model>, P: Borrow<SimplePlan<M>>> SimpleState<M, P>

pub fn new(plan: P) -> TractResult<SimpleState<M, P>>

pub fn reset_wires(&mut self) -> TractResult<()>

Reset wires state.

pub fn reset_op_states(&mut self) -> TractResult<()>

Reset wires state.

pub fn run(&mut self, inputs: TVec<Tensor>) -> TractResult<TVec<SharedTensor>>

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

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 compute_one(&mut self, node: usize) -> TractResult<()>

pub fn compute_recursively(&mut self, node: usize) -> TractResult<()>

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

pub fn model(&self) -> &Model

Trait Implementations

impl<M: Borrow<Model>, P: Borrow<SimplePlan<M>> + Clone> Clone for SimpleState<M, P>

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

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<M: Debug + Borrow<Model>, P: Debug + Borrow<SimplePlan<M>>> Debug for SimpleState<M, P>

fn fmt(&self, f: &mut Formatter) -> Result

Formats the value using the given formatter.