[−][src]Trait tract_core::model::ModelDsl

pub trait ModelDsl<TI, O> where
    TI: Fact + Clone + 'static,
    O: Debug + Display + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static, {
    fn single_prec(&self, id: usize) -> TractResult<Option<&BaseNode<TI, O>>>;

    fn single_prec_at(
        &self, 
        id: usize, 
        count: usize
    ) -> TractResult<Option<&BaseNode<TI, O>>>;

    fn single_succ(&self, id: usize) -> TractResult<Option<&BaseNode<TI, O>>>;

    fn single_succ_at(
        &self, 
        id: usize, 
        count: usize
    ) -> TractResult<Option<&BaseNode<TI, O>>>;
}

Extensions on ModelImpl to explore and build graph models more easily.

Required methods

fn single_prec(&self, id: usize) -> TractResult<Option<&BaseNode<TI, O>>>

Find the lone precursor of a node, if applicable.

fn single_prec_at(
    &self,
    id: usize,
    count: usize
) -> TractResult<Option<&BaseNode<TI, O>>>

Find the count-th precursor of a node id in a chain of single tensor operation, if applicable.

fn single_succ(&self, id: usize) -> TractResult<Option<&BaseNode<TI, O>>>

Find the lone succesor of a node, if applicable.

fn single_succ_at(
    &self,
    id: usize,
    count: usize
) -> TractResult<Option<&BaseNode<TI, O>>>

Find the count-th successor of a node id in a chain of single tensor operation, if applicable.

Loading content...

Implementors

impl<TI, O> ModelDsl<TI, O> for ModelImpl<TI, O> where
    TI: Fact + Clone + 'static,
    O: Debug + Display + AsRef<dyn Op> + AsMut<dyn Op> + Clone + 'static, [src]

Loading content...