Trait tract_hir::internal::ElementWiseMiniOp
source · pub trait ElementWiseMiniOp: Debug + DynClone + Send + Sync + 'static + Downcast {
// Required method
fn name(&self) -> String;
// 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,
out_dt: Option<DatumType>
) -> Result<(), Error> { ... }
fn eval_out_of_place(
&self,
t: &Tensor,
out_dt: Option<DatumType>
) -> Result<Tensor, Error> { ... }
fn cost_per_element(&self, dt: DatumType) -> SmallVec<[(Cost, usize); 4]> { ... }
fn operating_datum_type(&self, dt: DatumType) -> DatumType { ... }
fn declutter(
&self,
model: &Graph<TypedFact, Box<dyn TypedOp>>,
node: &Node<TypedFact, Box<dyn TypedOp>>
) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp>>>, Error> { ... }
fn quantize(
&self,
dt: DatumType,
scale: f32,
zero_point: i32
) -> Result<Option<Box<dyn ElementWiseMiniOp>>, Error> { ... }
fn info(&self) -> Result<Vec<String>, Error> { ... }
fn same_as(&self, other: &(dyn ElementWiseMiniOp + 'static)) -> bool { ... }
}
Required Methods§
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, out_dt: Option<DatumType> ) -> Result<(), Error>
fn eval_out_of_place( &self, t: &Tensor, out_dt: Option<DatumType> ) -> Result<Tensor, Error>
fn cost_per_element(&self, dt: DatumType) -> SmallVec<[(Cost, usize); 4]>
fn operating_datum_type(&self, dt: DatumType) -> DatumType
fn declutter( &self, model: &Graph<TypedFact, Box<dyn TypedOp>>, node: &Node<TypedFact, Box<dyn TypedOp>> ) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp>>>, Error>
fn quantize( &self, dt: DatumType, scale: f32, zero_point: i32 ) -> Result<Option<Box<dyn ElementWiseMiniOp>>, Error>
fn info(&self) -> Result<Vec<String>, Error>
fn same_as(&self, other: &(dyn ElementWiseMiniOp + 'static)) -> bool
Implementations§
source§impl dyn ElementWiseMiniOp
impl dyn ElementWiseMiniOp
sourcepub fn is<__T>(&self) -> boolwhere
__T: ElementWiseMiniOp,
pub fn is<__T>(&self) -> boolwhere
__T: ElementWiseMiniOp,
Returns true if the trait object wraps an object of type __T
.
sourcepub fn downcast<__T>(
self: Box<dyn ElementWiseMiniOp>
) -> Result<Box<__T>, Box<dyn ElementWiseMiniOp>>where
__T: ElementWiseMiniOp,
pub fn downcast<__T>(
self: Box<dyn ElementWiseMiniOp>
) -> Result<Box<__T>, Box<dyn ElementWiseMiniOp>>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.
sourcepub fn downcast_rc<__T>(
self: Rc<dyn ElementWiseMiniOp>
) -> Result<Rc<__T>, Rc<dyn ElementWiseMiniOp>>where
__T: ElementWiseMiniOp,
pub fn downcast_rc<__T>(
self: Rc<dyn ElementWiseMiniOp>
) -> Result<Rc<__T>, Rc<dyn ElementWiseMiniOp>>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.
sourcepub fn downcast_ref<__T>(&self) -> Option<&__T>where
__T: ElementWiseMiniOp,
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.
sourcepub fn downcast_mut<__T>(&mut self) -> Option<&mut __T>where
__T: ElementWiseMiniOp,
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.