Trait tract_hir::internal::ElementWiseMiniOp[−][src]

pub trait ElementWiseMiniOp: 'static + Debug + DynClone + Send + Sync + 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) -> Result<(), Error> { ... }

    fn eval_out_of_place(&self, t: &Tensor) -> Result<Tensor, Error> { ... }

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

    fn declutter(
        &self, 
        model: &Graph<TypedFact, Box<dyn TypedOp + 'static, Global>>, 
        node: &Node<TypedFact, Box<dyn TypedOp + 'static, Global>>
    ) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp + 'static, Global>>>, Error> { ... }

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

    fn info(&self) -> Result<Vec<String, Global>, Error> { ... }
}

Required methods

Provided methods

Implementations

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

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.

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.

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

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

Implementors