Struct tract_pulse::internal::tract_core::ops::cast::Cast
pub struct Cast {
pub to: DatumType,
}
Fields§
§to: DatumType
Implementations§
Trait Implementations§
§impl Op for Cast
impl Op for Cast
§impl OpStateFreeze for Cast
impl OpStateFreeze for Cast
fn freeze(&self) -> Box<dyn FrozenOpState, Global>
§impl TypedOp for Cast
impl TypedOp for Cast
§fn output_facts(
&self,
inputs: &[&TypedFact]
) -> Result<SmallVec<[TypedFact; 4]>, Error>
fn output_facts( &self, inputs: &[&TypedFact] ) -> Result<SmallVec<[TypedFact; 4]>, Error>
Deduce output facts from input facts.
§fn declutter(
&self,
model: &Graph<TypedFact, Box<dyn TypedOp, Global>>,
node: &Node<TypedFact, Box<dyn TypedOp, Global>>
) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp, Global>>>, Error>
fn declutter( &self, model: &Graph<TypedFact, Box<dyn TypedOp, Global>>, node: &Node<TypedFact, Box<dyn TypedOp, Global>> ) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp, Global>>>, Error>
Declutter the op to the tract_core operator set as much as possible.
fn axes_mapping( &self, inputs: &[&TypedFact], outputs: &[&TypedFact] ) -> Result<AxesMapping, Error>
fn change_axes( &self, model: &Graph<TypedFact, Box<dyn TypedOp, Global>>, node: &Node<TypedFact, Box<dyn TypedOp, Global>>, _io: InOut, change: &AxisOp ) -> Result<Option<AxisChangeConsequence>, Error>
§fn fuse(
&self,
_model: &Graph<TypedFact, Box<dyn TypedOp, Global>>,
_node: &Node<TypedFact, Box<dyn TypedOp, Global>>
) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp, Global>>>, Error>
fn fuse( &self, _model: &Graph<TypedFact, Box<dyn TypedOp, Global>>, _node: &Node<TypedFact, Box<dyn TypedOp, Global>> ) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp, Global>>>, Error>
Fuse op after codegen to deal with local optimisations.
§fn declutter_with_session(
&self,
session: &mut OptimizerSession<'_>,
model: &Graph<TypedFact, Box<dyn TypedOp, Global>>,
node: &Node<TypedFact, Box<dyn TypedOp, Global>>
) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp, Global>>>, Error>
fn declutter_with_session( &self, session: &mut OptimizerSession<'_>, model: &Graph<TypedFact, Box<dyn TypedOp, Global>>, node: &Node<TypedFact, Box<dyn TypedOp, Global>> ) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp, Global>>>, Error>
Declutter the op to the tract_core operator set as much as possible.
§fn cost(
&self,
_inputs: &[&TypedFact]
) -> Result<SmallVec<[(Cost, TDim); 4]>, Error>
fn cost( &self, _inputs: &[&TypedFact] ) -> Result<SmallVec<[(Cost, TDim); 4]>, Error>
Computes a cost hint of the operation. Read more
fn suggested_axis_changes( &self ) -> Result<SmallVec<[(InOut, AxisOp); 4]>, Error>
fn slice( &self, patch: &mut ModelPatch<TypedFact, Box<dyn TypedOp, Global>>, prefix: &str, inputs: &[OutletId], output_axis: usize, start: usize, end: usize ) -> Result<Option<SmallVec<[OutletId; 4]>>, Error>
§fn quantize(
&self,
model: &Graph<TypedFact, Box<dyn TypedOp, Global>>,
node: &Node<TypedFact, Box<dyn TypedOp, Global>>,
dt: DatumType,
scale: f32,
zero_point: i32
) -> Result<Option<Box<dyn TypedOp, Global>>, Error>
fn quantize( &self, model: &Graph<TypedFact, Box<dyn TypedOp, Global>>, node: &Node<TypedFact, Box<dyn TypedOp, Global>>, dt: DatumType, scale: f32, zero_point: i32 ) -> Result<Option<Box<dyn TypedOp, Global>>, Error>
Transforms the op in an equivalent one, operating on dt (i8 or u8). Read more
§fn concretize_dims(
&self,
source: &Graph<TypedFact, Box<dyn TypedOp, Global>>,
node: &Node<TypedFact, Box<dyn TypedOp, Global>>,
target: &mut Graph<TypedFact, Box<dyn TypedOp, Global>>,
mapping: &HashMap<OutletId, OutletId, RandomState>,
values: &SymbolValues
) -> Result<SmallVec<[OutletId; 4]>, Error>
fn concretize_dims( &self, source: &Graph<TypedFact, Box<dyn TypedOp, Global>>, node: &Node<TypedFact, Box<dyn TypedOp, Global>>, target: &mut Graph<TypedFact, Box<dyn TypedOp, Global>>, mapping: &HashMap<OutletId, OutletId, RandomState>, values: &SymbolValues ) -> Result<SmallVec<[OutletId; 4]>, Error>
Transform the op into by providing a value to one or more symbols.
Auto Trait Implementations§
impl RefUnwindSafe for Cast
impl Send for Cast
impl Sync for Cast
impl Unpin for Cast
impl UnwindSafe for Cast
Blanket Implementations§
source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere T: ?Sized,
source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
source§impl<T> Downcast for Twhere
T: Any,
impl<T> Downcast for Twhere T: Any,
source§fn into_any(self: Box<T, Global>) -> Box<dyn Any, Global>
fn into_any(self: Box<T, Global>) -> Box<dyn Any, Global>
Convert
Box<dyn Trait>
(where Trait: Downcast
) to Box<dyn Any>
. Box<dyn Any>
can
then be further downcast
into Box<ConcreteType>
where ConcreteType
implements Trait
.source§fn into_any_rc(self: Rc<T, Global>) -> Rc<dyn Any, Global>
fn into_any_rc(self: Rc<T, Global>) -> Rc<dyn Any, Global>
Convert
Rc<Trait>
(where Trait: Downcast
) to Rc<Any>
. Rc<Any>
can then be
further downcast
into Rc<ConcreteType>
where ConcreteType
implements Trait
.source§fn as_any(&self) -> &(dyn Any + 'static)
fn as_any(&self) -> &(dyn Any + 'static)
Convert
&Trait
(where Trait: Downcast
) to &Any
. This is needed since Rust cannot
generate &Any
’s vtable from &Trait
’s.source§fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)
fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)
Convert
&mut Trait
(where Trait: Downcast
) to &Any
. This is needed since Rust cannot
generate &mut Any
’s vtable from &mut Trait
’s.