Struct tract_core::plan::SimplePlan

pub struct SimplePlan<F, O, M>where
    F: Fact + Hash + Clone + 'static,
    O: Debug + Display + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static + Hash,
    M: Borrow<Graph<F, O>> + Hash,{
    pub model: M,
    pub outputs: Vec<OutletId>,
    pub order: Vec<usize>,
    pub flush_lists: Vec<TVec<usize>>,
    pub has_unresolved_symbols: bool,
    /* private fields */
}

Fields

model: M

outputs: Vec<OutletId>

order: Vec<usize>

flush_lists: Vec<TVec<usize>>

has_unresolved_symbols: bool

Implementations

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

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 new_for_outputs_and_deps(
    model: M,
    outputs: &[OutletId],
    deps: &[(usize, usize)]
) -> TractResult<SimplePlan<F, O, M>>

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

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

Trait Implementations

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

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 + Hash + Clone + 'static,
    O: Debug + Display + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static + Hash,
    M: Borrow<Graph<F, O>> + Hash,

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

Formats the value using the given formatter.

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

fn hash<H: Hasher>(&self, state: &mut H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)where
    H: Hasher,

Feeds a slice of this type into the given Hasher.