pub enum AxisOp {
Add(usize),
Rm(usize),
Move(usize, usize),
Reshape(usize, SmallVec<[TDim; 4]>, SmallVec<[TDim; 4]>),
}
Variants§
Implementations§
source§impl AxisOp
impl AxisOp
pub fn canonical(&self) -> Cow<'_, AxisOp>
pub fn simplify(&self) -> SmallVec<[AxisOp; 4]>
pub fn transform_axis(&self, axis: usize) -> Option<usize>
pub fn merge_incoming_change( &self, change: &AxisOp ) -> Option<(Option<AxisOp>, Option<AxisOp>)>
pub fn change_shape_array<D>(
&self,
shape: &mut SmallVec<[D; 4]>,
broadcasting: bool
) -> Result<(), Error>where
D: DimLike,
pub fn change_shape( &self, shape: &mut ShapeFact, broadcasting: bool ) -> Result<(), Error>
pub fn change_tensor( &self, tensor: &mut Tensor, broadcasting: bool ) -> Result<(), Error>
pub fn change_view<D>( &self, view: &mut ArrayBase<ViewRepr<&D>, Dim<IxDynImpl>> ) -> Result<(), Error>
pub fn change_view_mut<D>( &self, view: &mut ArrayBase<ViewRepr<&mut D>, Dim<IxDynImpl>> ) -> Result<(), Error>
pub fn recip(&self) -> AxisOp
pub fn is_noop(&self) -> bool
pub fn only_shape(&self) -> bool
pub fn wire_split_axis( model: &mut Graph<TypedFact, Box<dyn TypedOp>>, name: impl ToString, outlet: OutletId, axis: usize, outer_dim: usize ) -> Result<SmallVec<[OutletId; 4]>, Error>
pub fn wire_collapse_axis( model: &mut Graph<TypedFact, Box<dyn TypedOp>>, name: impl ToString, outlet: OutletId, axis: usize ) -> Result<SmallVec<[OutletId; 4]>, Error>
Trait Implementations§
source§impl EvalOp for AxisOp
impl EvalOp for AxisOp
fn is_stateless(&self) -> bool
fn eval_with_session( &self, session: &SessionState, inputs: SmallVec<[TValue; 4]> ) -> Result<SmallVec<[TValue; 4]>, Error>
fn eval( &self, inputs: SmallVec<[TValue; 4]> ) -> Result<SmallVec<[TValue; 4]>, Error>
fn state( &self, session: &mut SessionState, node_id: usize ) -> Result<Option<Box<dyn OpState>>, Error>
source§impl Op for AxisOp
impl Op for AxisOp
fn name(&self) -> Cow<'_, str>
source§fn info(&self) -> Result<Vec<String>, Error>
fn info(&self) -> Result<Vec<String>, Error>
Short (one-line) strings giving hints on internal implementation or
important configuration details to be displayed in dumps.
fn as_typed(&self) -> Option<&(dyn TypedOp + 'static)>
source§fn validation(&self) -> Validation
fn validation(&self) -> Validation
The kind of accuracy check that should be performed on operation when
testing them.
source§impl PartialEq for AxisOp
impl PartialEq for AxisOp
source§impl TypedOp for AxisOp
impl TypedOp for AxisOp
source§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 axes_mapping( &self, inputs: &[&TypedFact], outputs: &[&TypedFact] ) -> Result<AxesMapping, Error>
source§fn declutter(
&self,
model: &Graph<TypedFact, Box<dyn TypedOp>>,
node: &Node<TypedFact, Box<dyn TypedOp>>
) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp>>>, Error>
fn declutter( &self, model: &Graph<TypedFact, Box<dyn TypedOp>>, node: &Node<TypedFact, Box<dyn TypedOp>> ) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp>>>, Error>
Declutter the op to the tract_core operator set as much as possible.
fn suggested_axis_changes( &self ) -> Result<SmallVec<[(InOut, AxisOp); 4]>, Error>
fn change_axes( &self, _model: &Graph<TypedFact, Box<dyn TypedOp>>, _node: &Node<TypedFact, Box<dyn TypedOp>>, io: InOut, change: &AxisOp ) -> Result<Option<AxisChangeConsequence>, Error>
source§fn concretize_dims(
&self,
_source: &Graph<TypedFact, Box<dyn TypedOp>>,
node: &Node<TypedFact, Box<dyn TypedOp>>,
target: &mut Graph<TypedFact, Box<dyn TypedOp>>,
mapping: &HashMap<OutletId, OutletId>,
values: &SymbolValues
) -> Result<SmallVec<[OutletId; 4]>, Error>
fn concretize_dims( &self, _source: &Graph<TypedFact, Box<dyn TypedOp>>, node: &Node<TypedFact, Box<dyn TypedOp>>, target: &mut Graph<TypedFact, Box<dyn TypedOp>>, mapping: &HashMap<OutletId, OutletId>, values: &SymbolValues ) -> Result<SmallVec<[OutletId; 4]>, Error>
Transform the op into by providing a value to one or more symbols.
source§fn fuse(
&self,
_model: &Graph<TypedFact, Box<dyn TypedOp>>,
_node: &Node<TypedFact, Box<dyn TypedOp>>
) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp>>>, Error>
fn fuse( &self, _model: &Graph<TypedFact, Box<dyn TypedOp>>, _node: &Node<TypedFact, Box<dyn TypedOp>> ) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp>>>, Error>
Fuse op after codegen to deal with local optimisations.
source§fn declutter_with_session(
&self,
session: &mut OptimizerSession<'_>,
model: &Graph<TypedFact, Box<dyn TypedOp>>,
node: &Node<TypedFact, Box<dyn TypedOp>>
) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp>>>, Error>
fn declutter_with_session( &self, session: &mut OptimizerSession<'_>, model: &Graph<TypedFact, Box<dyn TypedOp>>, node: &Node<TypedFact, Box<dyn TypedOp>> ) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp>>>, Error>
Declutter the op to the tract_core operator set as much as possible.
source§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 slice( &self, patch: &mut ModelPatch<TypedFact, Box<dyn TypedOp>>, prefix: &str, inputs: &[OutletId], output_axis: usize, start: usize, end: usize ) -> Result<Option<SmallVec<[OutletId; 4]>>, Error>
source§fn quantize(
&self,
model: &Graph<TypedFact, Box<dyn TypedOp>>,
node: &Node<TypedFact, Box<dyn TypedOp>>,
dt: DatumType,
scale: f32,
zero_point: i32
) -> Result<Option<Box<dyn TypedOp>>, Error>
fn quantize( &self, model: &Graph<TypedFact, Box<dyn TypedOp>>, node: &Node<TypedFact, Box<dyn TypedOp>>, dt: DatumType, scale: f32, zero_point: i32 ) -> Result<Option<Box<dyn TypedOp>>, Error>
Transforms the op in an equivalent one, operating on dt (i8 or u8). Read more
impl Eq for AxisOp
Auto Trait Implementations§
impl Freeze for AxisOp
impl RefUnwindSafe for AxisOp
impl Send for AxisOp
impl Sync for AxisOp
impl Unpin for AxisOp
impl UnwindSafe for AxisOp
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>) -> Box<dyn Any>
fn into_any(self: Box<T>) -> Box<dyn Any>
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>) -> Rc<dyn Any>
fn into_any_rc(self: Rc<T>) -> Rc<dyn Any>
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.source§impl<T> DowncastSync for T
impl<T> DowncastSync for T
source§impl<Q, K> Equivalent<K> for Q
impl<Q, K> Equivalent<K> for Q
source§impl<T> IntoEither for T
impl<T> IntoEither for T
source§fn into_either(self, into_left: bool) -> Either<Self, Self> ⓘ
fn into_either(self, into_left: bool) -> Either<Self, Self> ⓘ
Converts
self
into a Left
variant of Either<Self, Self>
if into_left
is true
.
Converts self
into a Right
variant of Either<Self, Self>
otherwise. Read moresource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self> ⓘ
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self> ⓘ
Converts
self
into a Left
variant of Either<Self, Self>
if into_left(&self)
returns true
.
Converts self
into a Right
variant of Either<Self, Self>
otherwise. Read more