Trait tract_hir::internal::StatelessOp

pub trait StatelessOp: Op {
    fn eval(
        &self, 
        inputs: SmallVec<[Arc<Tensor>; 4]>
    ) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>;
}

Required methods

fn eval(
    &self,
    inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

Loading content...

Implementations on Foreign Types

impl<T> StatelessOp for MatMatMulPackB<T> where
    T: Copy + Datum + Zero, 

fn eval(
    &self,
    inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

impl StatelessOp for MergeOpUnicast

fn eval(
    &self,
    inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

impl StatelessOp for FiniteReshape

fn eval(
    &self,
    inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

impl StatelessOp for MultiBroadcastTo

fn eval(
    &self,
    inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

impl<T> StatelessOp for Im2Col<T> where
    T: Datum + Copy + Zero, 

fn eval(
    &self,
    inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

impl StatelessOp for UnaryOp

fn eval(
    &self,
    inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

impl StatelessOp for Reduce

fn eval(
    &self,
    inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

impl StatelessOp for MatMulUnary

fn eval(
    &self,
    inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

impl StatelessOp for Tile

fn eval(
    &self,
    inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

impl StatelessOp for TypedBinOp

fn eval(
    &self,
    inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

Loading content...

Implementors

impl StatelessOp for AxisOp

fn eval(
    &self,
    inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

impl StatelessOp for Box<dyn Expansion>

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

impl StatelessOp for ConstantLike

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

Evaluates the operation given the input tensors.

impl StatelessOp for EyeLike

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

Evaluates the operation given the input tensors.

impl StatelessOp for Gather

fn eval(
    &self,
    inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

Evaluates the operation given the input tensors.

impl StatelessOp for Pad

fn eval(
    &self,
    inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

Evaluates the operation given the input tensors.

impl StatelessOp for TypedConcat

fn eval(
    &self,
    inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

Evaluates the operation given the input tensors.

impl StatelessOp for Nary

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

impl StatelessOp for ConvUnary

fn eval(
    &self,
    inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

impl StatelessOp for MaxPool

fn eval(
    &self,
    inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

impl StatelessOp for SumPool

fn eval(
    &self,
    inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

impl StatelessOp for Downsample

fn eval(
    &self,
    inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

impl StatelessOp for Dummy

fn eval(
    &self,
    _inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

impl StatelessOp for ElementWiseOp

fn eval(
    &self,
    inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

impl StatelessOp for Identity

fn eval(
    &self,
    inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

Evaluates the operation given the input tensors.

impl StatelessOp for Const

fn eval(
    &self,
    _inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

impl StatelessOp for Iff

fn eval(
    &self,
    inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

impl StatelessOp for MatMul

fn eval(
    &self,
    inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

impl StatelessOp for LayerHardmax

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

impl StatelessOp for DequantizeLinearF32

fn eval(
    &self,
    inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

impl<D> StatelessOp for Slice<D> where
    D: DimLike + ToDim + Hash, 

fn eval(
    &self,
    inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>

Evaluates the operation given the input tensors.

Loading content...