Trait tract_core::ops::EvalOp

pub trait EvalOp {
    fn is_stateless(&self) -> bool;

    fn eval(&self, inputs: TVec<Arc<Tensor>>) -> TractResult<TVec<Arc<Tensor>>> { ... }
    fn state(
        &self, 
        session: &mut SessionState, 
        node_id: usize
    ) -> TractResult<Option<Box<dyn OpState>>> { ... }
}

Required methods

fn is_stateless(&self) -> bool

Loading content...

Provided methods

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

fn state(
    &self,
    session: &mut SessionState,
    node_id: usize
) -> TractResult<Option<Box<dyn OpState>>>

Loading content...

Implementors

impl EvalOp for AxisOp

fn is_stateless(&self) -> bool

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

impl EvalOp for ConstantOfShape

fn eval(&self, _inputs: TVec<Arc<Tensor>>) -> TractResult<TVec<Arc<Tensor>>>

fn is_stateless(&self) -> bool

fn state(
    &self,
    _session: &mut SessionState,
    _node_id: usize
) -> TractResult<Option<Box<dyn OpState>>>

impl EvalOp for FiniteReshape

fn is_stateless(&self) -> bool

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

impl EvalOp for Gather

fn is_stateless(&self) -> bool

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

impl EvalOp for MultiBroadcastTo

fn is_stateless(&self) -> bool

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

impl EvalOp for Pad

fn is_stateless(&self) -> bool

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

impl EvalOp for Tile

fn is_stateless(&self) -> bool

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

impl EvalOp for TypedConcat

fn is_stateless(&self) -> bool

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

impl EvalOp for MergeOpUnicast

fn is_stateless(&self) -> bool

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

impl EvalOp for TypedBinOp

fn is_stateless(&self) -> bool

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

impl EvalOp for UnaryOp

fn is_stateless(&self) -> bool

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

impl EvalOp for Im2Col

fn is_stateless(&self) -> bool

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

impl EvalOp for ConvUnary

fn is_stateless(&self) -> bool

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

impl EvalOp for MaxPool

fn is_stateless(&self) -> bool

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

impl EvalOp for SumPool

fn is_stateless(&self) -> bool

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

impl EvalOp for Downsample

fn is_stateless(&self) -> bool

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

impl EvalOp for Dummy

fn is_stateless(&self) -> bool

fn eval(&self, _inputs: TVec<Arc<Tensor>>) -> TractResult<TVec<Arc<Tensor>>>

impl EvalOp for ElementWiseOp

fn is_stateless(&self) -> bool

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

impl EvalOp for Identity

fn is_stateless(&self) -> bool

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

Evaluates the operation given the input tensors.

impl EvalOp for Const

fn is_stateless(&self) -> bool

fn eval(&self, _inputs: TVec<Arc<Tensor>>) -> TractResult<TVec<Arc<Tensor>>>

impl EvalOp for Iff

fn is_stateless(&self) -> bool

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

impl EvalOp for MatMul

fn is_stateless(&self) -> bool

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

impl EvalOp for MatMulUnary

fn is_stateless(&self) -> bool

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

impl EvalOp for MatMatMulPackB

fn is_stateless(&self) -> bool

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

impl EvalOp for Reduce

fn is_stateless(&self) -> bool

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

impl EvalOp for DequantizeLinearF32

fn is_stateless(&self) -> bool

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

impl EvalOp for LirScan

fn is_stateless(&self) -> bool

fn state(
    &self,
    _session: &mut SessionState,
    _node_id: usize
) -> TractResult<Option<Box<dyn OpState>>>

impl EvalOp for Scan

fn is_stateless(&self) -> bool

fn state(
    &self,
    session: &mut SessionState,
    node_id: usize
) -> TractResult<Option<Box<dyn OpState>>>

impl EvalOp for TypedSource

fn is_stateless(&self) -> bool

fn state(
    &self,
    _session: &mut SessionState,
    node_id: usize
) -> TractResult<Option<Box<dyn OpState>>>

impl EvalOp for UnimplementedOp

fn is_stateless(&self) -> bool

impl<D: DimLike + ToDim + Hash> EvalOp for Slice<D>

fn is_stateless(&self) -> bool

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

Loading content...