Trait tract_hir::internal::OpState

pub trait OpState: Send + Debug {
    fn eval(
        &mut self, 
        session: &mut SessionState, 
        op: &(dyn Op + 'static), 
        inputs: SmallVec<[Arc<Tensor>; 4]>
    ) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>;
}

Required methods

fn eval(
    &mut self,
    session: &mut SessionState,
    op: &(dyn Op + 'static),
    inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

Loading content...

Implementations on Foreign Types

impl OpState for SourceState

fn eval(
    &mut self,
    session: &mut SessionState,
    _op: &(dyn Op + 'static),
    _inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

Loading content...

Implementors

Loading content...