Trait tract_core::ops::element_wise::ElementWiseMiniOp

pub trait ElementWiseMiniOp: Debug + DynClone + Send + Sync + 'static + Downcast + DynHash {
    fn name(&self) -> String;

    fn prefix(&self) -> &'static str { ... }
    fn validation(&self) -> Validation { ... }
    fn output_type(&self, input_type: DatumType) -> Option<DatumType> { ... }
    fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()> { ... }
    fn eval_out_of_place(&self, t: &Tensor) -> TractResult<Tensor> { ... }
    fn cost_per_element(&self, dt: DatumType) -> TVec<(Cost, usize)> { ... }
    fn declutter(
        &self, 
        model: &TypedModel, 
        node: &TypedNode
    ) -> TractResult<Option<TypedModelPatch>> { ... }
    fn quantize(
        &self, 
        dt: DatumType, 
        scale: f32, 
        zero_point: i32
    ) -> TractResult<Option<Box<dyn ElementWiseMiniOp>>> { ... }
    fn info(&self) -> TractResult<Vec<String>> { ... }
}

Required methods

fn name(&self) -> String

Loading content...

Provided methods

fn prefix(&self) -> &'static str

fn validation(&self) -> Validation

fn output_type(&self, input_type: DatumType) -> Option<DatumType>

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

fn eval_out_of_place(&self, t: &Tensor) -> TractResult<Tensor>

fn cost_per_element(&self, dt: DatumType) -> TVec<(Cost, usize)>

fn declutter(
    &self,
    model: &TypedModel,
    node: &TypedNode
) -> TractResult<Option<TypedModelPatch>>

fn quantize(
    &self,
    dt: DatumType,
    scale: f32,
    zero_point: i32
) -> TractResult<Option<Box<dyn ElementWiseMiniOp>>>

fn info(&self) -> TractResult<Vec<String>>

Loading content...

Implementations

impl dyn ElementWiseMiniOp

pub fn is<__T: ElementWiseMiniOp>(&self) -> bool

Returns true if the trait object wraps an object of type __T.

pub fn downcast<__T: ElementWiseMiniOp>(
    self: Box<Self>
) -> Result<Box<__T>, Box<Self>>

Returns a boxed object from a boxed trait object if the underlying object is of type __T. Returns the original boxed trait if it isn't.

pub fn downcast_rc<__T: ElementWiseMiniOp>(
    self: Rc<Self>
) -> Result<Rc<__T>, Rc<Self>>

Returns an Rc-ed object from an Rc-ed trait object if the underlying object is of type __T. Returns the original Rc-ed trait if it isn't.

pub fn downcast_ref<__T: ElementWiseMiniOp>(&self) -> Option<&__T>

Returns a reference to the object within the trait object if it is of type __T, or None if it isn't.

pub fn downcast_mut<__T: ElementWiseMiniOp>(&mut self) -> Option<&mut __T>

Returns a mutable reference to the object within the trait object if it is of type __T, or None if it isn't.

Implementors

impl ElementWiseMiniOp for Cast

fn name(&self) -> String

fn output_type(&self, _input_type: DatumType) -> Option<DatumType>

fn eval_out_of_place(&self, t: &Tensor) -> TractResult<Tensor>

fn declutter(
    &self,
    model: &TypedModel,
    node: &TypedNode
) -> TractResult<Option<TypedModelPatch>>

impl ElementWiseMiniOp for Not

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

impl ElementWiseMiniOp for Abs

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

impl ElementWiseMiniOp for Acos

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

impl ElementWiseMiniOp for Acosh

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

impl ElementWiseMiniOp for Asin

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

impl ElementWiseMiniOp for Asinh

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

impl ElementWiseMiniOp for Atan

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

impl ElementWiseMiniOp for Atanh

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

impl ElementWiseMiniOp for Ceil

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

impl ElementWiseMiniOp for Cos

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

impl ElementWiseMiniOp for Cosh

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

impl ElementWiseMiniOp for Exp

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

fn validation(&self) -> Validation

impl ElementWiseMiniOp for Floor

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

impl ElementWiseMiniOp for Ln

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

fn validation(&self) -> Validation

impl ElementWiseMiniOp for Neg

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

impl ElementWiseMiniOp for Recip

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

fn cost_per_element(&self, dt: DatumType) -> TVec<(Cost, usize)>

fn validation(&self) -> Validation

impl ElementWiseMiniOp for Round

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

impl ElementWiseMiniOp for Rsqrt

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

fn validation(&self) -> Validation

impl ElementWiseMiniOp for ScalarMax

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

fn quantize(
    &self,
    dt: DatumType,
    scale: f32,
    zero_point: i32
) -> TractResult<Option<Box<dyn ElementWiseMiniOp>>>

impl ElementWiseMiniOp for ScalarMin

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

fn quantize(
    &self,
    dt: DatumType,
    scale: f32,
    zero_point: i32
) -> TractResult<Option<Box<dyn ElementWiseMiniOp>>>

impl ElementWiseMiniOp for ScalarMinMax

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

fn quantize(
    &self,
    dt: DatumType,
    scale: f32,
    zero_point: i32
) -> TractResult<Option<Box<dyn ElementWiseMiniOp>>>

impl ElementWiseMiniOp for Sign

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

impl ElementWiseMiniOp for Sin

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

impl ElementWiseMiniOp for Sinh

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

impl ElementWiseMiniOp for Sqrt

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

fn validation(&self) -> Validation

impl ElementWiseMiniOp for Square

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

fn validation(&self) -> Validation

impl ElementWiseMiniOp for Tan

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

impl ElementWiseMiniOp for Tanh

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

fn cost_per_element(&self, dt: DatumType) -> TVec<(Cost, usize)>

impl ElementWiseMiniOp for Sigmoid

fn name(&self) -> String

fn eval_in_place(&self, t: &mut Tensor) -> TractResult<()>

fn cost_per_element(&self, dt: DatumType) -> TVec<(Cost, usize)>

impl ElementWiseMiniOp for LookupTable

fn name(&self) -> String

fn output_type(&self, input_type: DatumType) -> Option<DatumType>

fn eval_out_of_place(&self, t: &Tensor) -> TractResult<Tensor>

impl ElementWiseMiniOp for QuantizeLinearI8

fn name(&self) -> String

fn output_type(&self, input_type: DatumType) -> Option<DatumType>

fn eval_out_of_place(&self, t: &Tensor) -> TractResult<Tensor>

fn info(&self) -> TractResult<Vec<String>>

impl ElementWiseMiniOp for QuantizeLinearU8

fn name(&self) -> String

fn output_type(&self, input_type: DatumType) -> Option<DatumType>

fn eval_out_of_place(&self, t: &Tensor) -> TractResult<Tensor>

fn info(&self) -> TractResult<Vec<String>>

Loading content...