Struct tract_core::plan::SimplePlan

pub struct SimplePlan<F, O, M> where
    F: Fact + Clone + 'static,
    O: Debug + Display + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static,
    M: Borrow<ModelImpl<F, O>>,  {
    pub model: M,
    pub outputs: Vec<OutletId>,
    pub order: Vec<usize>,
    pub flush_lists: Vec<TVec<usize>>,
    // some fields omitted
}

Fields

model: M

outputs: Vec<OutletId>

order: Vec<usize>

flush_lists: Vec<TVec<usize>>

Methods

impl<F, O, M> SimplePlan<F, O, M> where
    F: Fact + Clone + 'static,
    O: Debug + Display + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static,
    M: Borrow<ModelImpl<F, O>>, 

pub fn new(model: M) -> TractResult<SimplePlan<F, O, M>>

This contructor returns a plan that will compute all the model default outputs in one pass.

pub fn new_for_output(
    model: M,
    output: OutletId
) -> TractResult<SimplePlan<F, O, M>>

This contructor returns a plan that will compute the specified output.

pub fn new_for_outputs(
    model: M,
    outputs: &[OutletId]
) -> TractResult<SimplePlan<F, O, M>>

This contructor returns a plan that will compute all specified outputs in one pass.

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

pub fn model(&self) -> &ModelImpl<F, O>

Trait Implementations

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

fn clone(&self) -> SimplePlan<F, O, M>

Returns a copy of the value.

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

Performs copy-assignment from source.

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

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

Formats the value using the given formatter.