Trait tract_hir::internal::ElementWiseMiniOp

pub trait ElementWiseMiniOp: 'static + Send + Sync + Debug + Downcast + DynHash + DynClone {
    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) -> Result<(), TractError> { ... }
    fn eval_out_of_place(&self, t: &Tensor) -> Result<Tensor, TractError> { ... }
    fn cost_per_element(&self, dt: DatumType) -> SmallVec<[(Cost, usize); 4]> { ... }
    fn declutter(
        &self, 
        model: &Graph<TypedFact, Box<dyn TypedOp + 'static>>, 
        node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>
    ) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp + 'static>>>, TractError> { ... }
    fn quantize(
        &self, 
        dt: DatumType, 
        scale: f32, 
        zero_point: i32
    ) -> Result<Option<Box<dyn ElementWiseMiniOp + 'static>>, TractError> { ... }
    fn info(&self) -> Result<Vec<String>, TractError> { ... }
}

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) -> Result<(), TractError>

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

fn cost_per_element(&self, dt: DatumType) -> SmallVec<[(Cost, usize); 4]>

fn declutter(
    &self,
    model: &Graph<TypedFact, Box<dyn TypedOp + 'static>>,
    node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>
) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp + 'static>>>, TractError>

fn quantize(
    &self,
    dt: DatumType,
    scale: f32,
    zero_point: i32
) -> Result<Option<Box<dyn ElementWiseMiniOp + 'static>>, TractError>

fn info(&self) -> Result<Vec<String>, TractError>

Loading content...

Implementations

impl dyn ElementWiseMiniOp + 'static

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

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

pub fn downcast<__T>(
    self: Box<dyn ElementWiseMiniOp + 'static>
) -> Result<Box<__T>, Box<dyn ElementWiseMiniOp + 'static>> where
    __T: ElementWiseMiniOp, 

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>(
    self: Rc<dyn ElementWiseMiniOp + 'static>
) -> Result<Rc<__T>, Rc<dyn ElementWiseMiniOp + 'static>> where
    __T: ElementWiseMiniOp, 

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>(&self) -> Option<&__T> where
    __T: ElementWiseMiniOp, 

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>(&mut self) -> Option<&mut __T> where
    __T: ElementWiseMiniOp, 

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

Implementations on Foreign Types

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) -> Result<Tensor, TractError>

fn declutter(
    &self,
    model: &Graph<TypedFact, Box<dyn TypedOp + 'static>>,
    node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>
) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp + 'static>>>, TractError>

impl ElementWiseMiniOp for Sigmoid

fn name(&self) -> String

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

fn cost_per_element(&self, dt: DatumType) -> SmallVec<[(Cost, usize); 4]>

Loading content...

Implementors

impl ElementWiseMiniOp for Not

fn name(&self) -> String

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

impl ElementWiseMiniOp for Abs

fn name(&self) -> String

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

impl ElementWiseMiniOp for Acos

fn name(&self) -> String

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

impl ElementWiseMiniOp for Acosh

fn name(&self) -> String

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

impl ElementWiseMiniOp for Asin

fn name(&self) -> String

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

impl ElementWiseMiniOp for Asinh

fn name(&self) -> String

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

impl ElementWiseMiniOp for Atan

fn name(&self) -> String

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

impl ElementWiseMiniOp for Atanh

fn name(&self) -> String

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

impl ElementWiseMiniOp for Ceil

fn name(&self) -> String

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

impl ElementWiseMiniOp for Cos

fn name(&self) -> String

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

impl ElementWiseMiniOp for Cosh

fn name(&self) -> String

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

impl ElementWiseMiniOp for Exp

fn name(&self) -> String

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

fn validation(&self) -> Validation

impl ElementWiseMiniOp for Floor

fn name(&self) -> String

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

impl ElementWiseMiniOp for Ln

fn name(&self) -> String

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

fn validation(&self) -> Validation

impl ElementWiseMiniOp for Neg

fn name(&self) -> String

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

impl ElementWiseMiniOp for Recip

fn name(&self) -> String

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

fn cost_per_element(&self, dt: DatumType) -> SmallVec<[(Cost, usize); 4]>

fn validation(&self) -> Validation

impl ElementWiseMiniOp for Round

fn name(&self) -> String

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

impl ElementWiseMiniOp for RoundHalfToEven

fn name(&self) -> String

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

impl ElementWiseMiniOp for Rsqrt

fn name(&self) -> String

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

fn validation(&self) -> Validation

impl ElementWiseMiniOp for Sign

fn name(&self) -> String

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

impl ElementWiseMiniOp for Sin

fn name(&self) -> String

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

impl ElementWiseMiniOp for Sinh

fn name(&self) -> String

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

impl ElementWiseMiniOp for Sqrt

fn name(&self) -> String

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

fn validation(&self) -> Validation

impl ElementWiseMiniOp for Square

fn name(&self) -> String

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

fn validation(&self) -> Validation

impl ElementWiseMiniOp for Tan

fn name(&self) -> String

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

impl ElementWiseMiniOp for Tanh

fn name(&self) -> String

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

fn cost_per_element(&self, dt: DatumType) -> SmallVec<[(Cost, usize); 4]>

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) -> Result<Tensor, TractError>

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) -> Result<Tensor, TractError>

fn info(&self) -> Result<Vec<String>, TractError>

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) -> Result<Tensor, TractError>

fn info(&self) -> Result<Vec<String>, TractError>

Loading content...