pub trait BinMiniOp:
Debug
+ DynClone
+ DynEq
+ Send
+ Sync
+ 'static
+ Downcast {
Show 18 methods
// Required methods
fn name(&self) -> &'static str;
fn result_datum_type(
&self,
a: DatumType,
b: DatumType,
) -> TractResult<DatumType>;
fn eval_in_a(&self, a: &mut Tensor, b: &Tensor) -> TractResult<()>;
fn eval_out_of_place(
&self,
c: &mut Tensor,
a: &Tensor,
b: &Tensor,
) -> TractResult<()>;
// Provided methods
fn validation(&self) -> Validation { ... }
fn operating_datum_type(
&self,
a: DatumType,
b: DatumType,
) -> TractResult<DatumType> { ... }
fn is_commutative(&self) -> bool { ... }
fn neutral_element(&self) -> Option<i64> { ... }
fn absorbing_element(&self) -> Option<i64> { ... }
fn maybe_eval_qbinary_as_float_op(
&self,
a: &TValue,
b: &TValue,
c_dt: &DatumType,
) -> TractResult<Option<Tensor>> { ... }
fn generic_eval(
&self,
a: TValue,
b: TValue,
c_dt: DatumType,
) -> TractResult<Tensor> { ... }
fn eval(&self, a: TValue, b: TValue, c_dt: DatumType) -> TractResult<Tensor> { ... }
fn declutter(
&self,
model: &TypedModel,
node: &TypedNode,
) -> TractResult<Option<TypedModelPatch>> { ... }
fn codegen(
&self,
model: &TypedModel,
node: &TypedNode,
) -> TractResult<Option<TypedModelPatch>> { ... }
fn cost_per_element(&self, dt: DatumType) -> TVec<(Cost, usize)> { ... }
fn as_linalg_binop(&self) -> Option<BinOp> { ... }
fn eval_symbolic(
&self,
session: &TurnState,
inputs: TVec<TValue>,
) -> TractResult<Option<TVec<TValue>>> { ... }
fn uniform_tdim_comparison(&self, a: &TDim, b: &TDim) -> Option<TDim> { ... }
}Required Methods§
fn name(&self) -> &'static str
fn result_datum_type( &self, a: DatumType, b: DatumType, ) -> TractResult<DatumType>
fn eval_in_a(&self, a: &mut Tensor, b: &Tensor) -> TractResult<()>
fn eval_out_of_place( &self, c: &mut Tensor, a: &Tensor, b: &Tensor, ) -> TractResult<()>
Provided Methods§
fn validation(&self) -> Validation
fn operating_datum_type( &self, a: DatumType, b: DatumType, ) -> TractResult<DatumType>
fn is_commutative(&self) -> bool
fn neutral_element(&self) -> Option<i64>
fn absorbing_element(&self) -> Option<i64>
fn maybe_eval_qbinary_as_float_op( &self, a: &TValue, b: &TValue, c_dt: &DatumType, ) -> TractResult<Option<Tensor>>
fn generic_eval( &self, a: TValue, b: TValue, c_dt: DatumType, ) -> TractResult<Tensor>
fn eval(&self, a: TValue, b: TValue, c_dt: DatumType) -> TractResult<Tensor>
fn declutter( &self, model: &TypedModel, node: &TypedNode, ) -> TractResult<Option<TypedModelPatch>>
fn codegen( &self, model: &TypedModel, node: &TypedNode, ) -> TractResult<Option<TypedModelPatch>>
fn cost_per_element(&self, dt: DatumType) -> TVec<(Cost, usize)>
fn as_linalg_binop(&self) -> Option<BinOp>
Sourcefn eval_symbolic(
&self,
session: &TurnState,
inputs: TVec<TValue>,
) -> TractResult<Option<TVec<TValue>>>
fn eval_symbolic( &self, session: &TurnState, inputs: TVec<TValue>, ) -> TractResult<Option<TVec<TValue>>>
Override for ops that can evaluate symbolic TDim inputs (comparisons).
Implementations§
Source§impl dyn BinMiniOp
impl dyn BinMiniOp
Sourcepub fn is<__T: BinMiniOp>(&self) -> bool
pub fn is<__T: BinMiniOp>(&self) -> bool
Returns true if the trait object wraps an object of type __T.
Sourcepub fn downcast<__T: BinMiniOp>(self: Box<Self>) -> Result<Box<__T>, Box<Self>>
pub fn downcast<__T: BinMiniOp>(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.
Sourcepub fn downcast_rc<__T: BinMiniOp>(self: Rc<Self>) -> Result<Rc<__T>, Rc<Self>>
pub fn downcast_rc<__T: BinMiniOp>(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.
Sourcepub fn downcast_ref<__T: BinMiniOp>(&self) -> Option<&__T>
pub fn downcast_ref<__T: BinMiniOp>(&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.
Sourcepub fn downcast_mut<__T: BinMiniOp>(&mut self) -> Option<&mut __T>
pub fn downcast_mut<__T: BinMiniOp>(&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.