[−][src]Enum tract_hir::internal::AxisOp
Variants
Add(usize)
Rm(usize)
Implementations
impl AxisOp
[src]
pub fn transform_axis(&self, axis: usize) -> Option<usize>
[src]
pub fn merge_incoming_change(
&self,
change: &AxisOp
) -> Option<(Option<AxisOp>, Option<AxisOp>)>
[src]
&self,
change: &AxisOp
) -> Option<(Option<AxisOp>, Option<AxisOp>)>
pub fn change_shape_array<D>(&self, shape: &mut SmallVec<[D; 4]>) where
D: DimLike,
[src]
D: DimLike,
pub fn change_shape(&self, shape: &mut ShapeFact) -> Result<(), TractError>
[src]
pub fn change_tensor(&self, tensor: &mut Tensor) -> Result<(), TractError>
[src]
pub fn change_tensor_broadcast_aware(
&self,
tensor: &mut Tensor
) -> Result<(), TractError>
[src]
&self,
tensor: &mut Tensor
) -> Result<(), TractError>
pub fn recip(&self) -> AxisOp
[src]
pub fn is_noop(&self) -> bool
[src]
pub fn only_shape(&self) -> bool
[src]
Trait Implementations
impl Clone for AxisOp
[src]
impl Debug for AxisOp
[src]
impl DynHash for AxisOp
impl Hash for AxisOp
[src]
fn hash<__H>(&self, state: &mut __H) where
__H: Hasher,
[src]
__H: Hasher,
fn hash_slice<H>(data: &[Self], state: &mut H) where
H: Hasher,
1.3.0[src]
H: Hasher,
impl Op for AxisOp
[src]
fn name(&self) -> Cow<'_, str>
[src]
fn info(&self) -> Result<Vec<String>, TractError>
[src]
fn is_canonic(&self) -> bool
[src]
fn op_families(&self) -> &'static [&'static str]ⓘ
[src]
fn as_typed(&self) -> Option<&(dyn TypedOp + 'static)>
[src]
fn as_pulsed(&self) -> Option<&(dyn PulsedOp + 'static)>
[src]
fn validation(&self) -> Validation
[src]
fn same_as(&self, _other: &(dyn Op + 'static)) -> bool
[src]
impl PartialEq<AxisOp> for AxisOp
[src]
impl PulsedOp for AxisOp
[src]
fn pulsed_output_facts(
&self,
inputs: &[&PulsedFact]
) -> Result<SmallVec<[PulsedFact; 4]>, TractError>
[src]
&self,
inputs: &[&PulsedFact]
) -> Result<SmallVec<[PulsedFact; 4]>, TractError>
fn as_op(&self) -> &(dyn Op + 'static)
[src]
fn as_op_mut(&mut self) -> &mut (dyn Op + 'static)
[src]
fn to_typed(&self) -> Box<dyn TypedOp + 'static>ⓘNotable traits for Box<W>
impl<W> Write for Box<W> where
W: Write + ?Sized, impl<R> Read for Box<R> where
R: Read + ?Sized, impl<I> Iterator for Box<I> where
I: Iterator + ?Sized, type Item = <I as Iterator>::Item;impl<F> Future for Box<F> where
F: Unpin + Future + ?Sized, type Output = <F as Future>::Output;
[src]
Notable traits for Box<W>
impl<W> Write for Box<W> where
W: Write + ?Sized, impl<R> Read for Box<R> where
R: Read + ?Sized, impl<I> Iterator for Box<I> where
I: Iterator + ?Sized, type Item = <I as Iterator>::Item;impl<F> Future for Box<F> where
F: Unpin + Future + ?Sized, type Output = <F as Future>::Output;
impl StatelessOp for AxisOp
[src]
fn eval(
&self,
inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>
[src]
&self,
inputs: SmallVec<[Arc<Tensor>; 4]>
) -> Result<SmallVec<[Arc<Tensor>; 4]>, TractError>
impl TypedOp for AxisOp
[src]
fn as_op(&self) -> &(dyn Op + 'static)
[src]
fn as_op_mut(&mut self) -> &mut (dyn Op + 'static)
[src]
fn output_facts(
&self,
inputs: &[&TypedFact]
) -> Result<SmallVec<[TypedFact; 4]>, TractError>
[src]
&self,
inputs: &[&TypedFact]
) -> Result<SmallVec<[TypedFact; 4]>, TractError>
fn invariants(
&self,
_model: &Graph<TypedFact, Box<dyn TypedOp + 'static>>,
node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>
) -> Result<Invariants, TractError>
[src]
&self,
_model: &Graph<TypedFact, Box<dyn TypedOp + 'static>>,
node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>
) -> Result<Invariants, TractError>
fn suggested_axis_changes(
&self
) -> Result<SmallVec<[(InOut, AxisOp); 4]>, TractError>
[src]
&self
) -> Result<SmallVec<[(InOut, AxisOp); 4]>, TractError>
fn change_axes(
&self,
model: &Graph<TypedFact, Box<dyn TypedOp + 'static>>,
node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>,
io: InOut,
change: &AxisOp
) -> Result<Option<AxisChangeConsequence>, TractError>
[src]
&self,
model: &Graph<TypedFact, Box<dyn TypedOp + 'static>>,
node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>,
io: InOut,
change: &AxisOp
) -> Result<Option<AxisChangeConsequence>, TractError>
fn pulsify(
&self,
_source: &Graph<TypedFact, Box<dyn TypedOp + 'static>>,
node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>,
target: &mut Graph<PulsedFact, Box<dyn PulsedOp + 'static>>,
mapping: &HashMap<OutletId, OutletId, RandomState>,
_pulse: usize
) -> Result<SmallVec<[OutletId; 4]>, TractError>
[src]
&self,
_source: &Graph<TypedFact, Box<dyn TypedOp + 'static>>,
node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>,
target: &mut Graph<PulsedFact, Box<dyn PulsedOp + 'static>>,
mapping: &HashMap<OutletId, OutletId, RandomState>,
_pulse: usize
) -> Result<SmallVec<[OutletId; 4]>, TractError>
fn fuse(
&self,
_model: &Graph<TypedFact, Box<dyn TypedOp + 'static>>,
_node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>
) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp + 'static>>>, TractError>
[src]
&self,
_model: &Graph<TypedFact, Box<dyn TypedOp + 'static>>,
_node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>
) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp + 'static>>>, TractError>
fn declutter(
&self,
model: &Graph<TypedFact, Box<dyn TypedOp + 'static>>,
node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>
) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp + 'static>>>, TractError>
[src]
&self,
model: &Graph<TypedFact, Box<dyn TypedOp + 'static>>,
node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>
) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp + 'static>>>, TractError>
fn cost(
&self,
_inputs: &[&TypedFact]
) -> Result<SmallVec<[(Cost, TDim); 4]>, TractError>
[src]
&self,
_inputs: &[&TypedFact]
) -> Result<SmallVec<[(Cost, TDim); 4]>, TractError>
fn slice_output(
&self,
model: &Graph<TypedFact, Box<dyn TypedOp + 'static>>,
node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>,
patch: &mut ModelPatch<TypedFact, Box<dyn TypedOp + 'static>>,
output_slot: usize,
axis: usize,
start: usize,
end: usize
) -> Result<Option<OutletId>, TractError>
[src]
&self,
model: &Graph<TypedFact, Box<dyn TypedOp + 'static>>,
node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>,
patch: &mut ModelPatch<TypedFact, Box<dyn TypedOp + 'static>>,
output_slot: usize,
axis: usize,
start: usize,
end: usize
) -> Result<Option<OutletId>, TractError>
fn quantize(
&self,
model: &Graph<TypedFact, Box<dyn TypedOp + 'static>>,
node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>,
dt: DatumType,
scale: f32,
zero_point: i32
) -> Result<Option<Box<dyn TypedOp + 'static>>, TractError>
[src]
&self,
model: &Graph<TypedFact, Box<dyn TypedOp + 'static>>,
node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>,
dt: DatumType,
scale: f32,
zero_point: i32
) -> Result<Option<Box<dyn TypedOp + 'static>>, TractError>
fn concretize_stream_dim(
&self,
_source: &Graph<TypedFact, Box<dyn TypedOp + 'static>>,
node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>,
target: &mut Graph<TypedFact, Box<dyn TypedOp + 'static>>,
mapping: &HashMap<OutletId, OutletId, RandomState>,
_stream_dim: usize
) -> Result<SmallVec<[OutletId; 4]>, TractError>
[src]
&self,
_source: &Graph<TypedFact, Box<dyn TypedOp + 'static>>,
node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>,
target: &mut Graph<TypedFact, Box<dyn TypedOp + 'static>>,
mapping: &HashMap<OutletId, OutletId, RandomState>,
_stream_dim: usize
) -> Result<SmallVec<[OutletId; 4]>, TractError>
fn codegen(
&self,
_model: &Graph<TypedFact, Box<dyn TypedOp + 'static>>,
_node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>
) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp + 'static>>>, TractError>
[src]
&self,
_model: &Graph<TypedFact, Box<dyn TypedOp + 'static>>,
_node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>
) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp + 'static>>>, TractError>
fn nested_model_multipliers(
&self,
inputs: &[&TypedFact]
) -> Vec<(Cow<'_, str>, f64)>ⓘ
[src]
&self,
inputs: &[&TypedFact]
) -> Vec<(Cow<'_, str>, f64)>ⓘ
Auto Trait Implementations
impl RefUnwindSafe for AxisOp
impl Send for AxisOp
impl Sync for AxisOp
impl Unpin for AxisOp
impl UnwindSafe for AxisOp
Blanket Implementations
impl<T> Any for T where
T: 'static + ?Sized,
[src]
T: 'static + ?Sized,
impl<T> Borrow<T> for T where
T: ?Sized,
[src]
T: ?Sized,
fn borrow(&self) -> &TⓘNotable traits for &'_ mut W
impl<'_, W> Write for &'_ mut W where
W: Write + ?Sized, impl<'_, R> Read for &'_ mut R where
R: Read + ?Sized, impl<'_, F> Future for &'_ mut F where
F: Unpin + Future + ?Sized, type Output = <F as Future>::Output;impl<'_, I> Iterator for &'_ mut I where
I: Iterator + ?Sized, type Item = <I as Iterator>::Item;
[src]
Notable traits for &'_ mut W
impl<'_, W> Write for &'_ mut W where
W: Write + ?Sized, impl<'_, R> Read for &'_ mut R where
R: Read + ?Sized, impl<'_, F> Future for &'_ mut F where
F: Unpin + Future + ?Sized, type Output = <F as Future>::Output;impl<'_, I> Iterator for &'_ mut I where
I: Iterator + ?Sized, type Item = <I as Iterator>::Item;
impl<T> BorrowMut<T> for T where
T: ?Sized,
[src]
T: ?Sized,
fn borrow_mut(&mut self) -> &mut TⓘNotable traits for &'_ mut W
impl<'_, W> Write for &'_ mut W where
W: Write + ?Sized, impl<'_, R> Read for &'_ mut R where
R: Read + ?Sized, impl<'_, F> Future for &'_ mut F where
F: Unpin + Future + ?Sized, type Output = <F as Future>::Output;impl<'_, I> Iterator for &'_ mut I where
I: Iterator + ?Sized, type Item = <I as Iterator>::Item;
[src]
Notable traits for &'_ mut W
impl<'_, W> Write for &'_ mut W where
W: Write + ?Sized, impl<'_, R> Read for &'_ mut R where
R: Read + ?Sized, impl<'_, F> Future for &'_ mut F where
F: Unpin + Future + ?Sized, type Output = <F as Future>::Output;impl<'_, I> Iterator for &'_ mut I where
I: Iterator + ?Sized, type Item = <I as Iterator>::Item;
impl<T> Downcast for T where
T: Any,
T: Any,
fn into_any(self: Box<T>) -> Box<dyn Any + 'static>ⓘNotable traits for Box<W>
impl<W> Write for Box<W> where
W: Write + ?Sized, impl<R> Read for Box<R> where
R: Read + ?Sized, impl<I> Iterator for Box<I> where
I: Iterator + ?Sized, type Item = <I as Iterator>::Item;impl<F> Future for Box<F> where
F: Unpin + Future + ?Sized, type Output = <F as Future>::Output;
Notable traits for Box<W>
impl<W> Write for Box<W> where
W: Write + ?Sized, impl<R> Read for Box<R> where
R: Read + ?Sized, impl<I> Iterator for Box<I> where
I: Iterator + ?Sized, type Item = <I as Iterator>::Item;impl<F> Future for Box<F> where
F: Unpin + Future + ?Sized, type Output = <F as Future>::Output;
fn into_any_rc(self: Rc<T>) -> Rc<dyn Any + 'static>
fn as_any(&self) -> &(dyn Any + 'static)
fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)
impl<T> DowncastSync for T where
T: Send + Sync + Any,
T: Send + Sync + Any,
impl<T> DynClone for T where
T: Clone,
[src]
T: Clone,
impl<T> From<T> for T
[src]
impl<T, U> Into<U> for T where
U: From<T>,
[src]
U: From<T>,
impl<O> StatefullOp for O where
O: StatelessOp + Clone,
[src]
O: StatelessOp + Clone,
fn state(
&self,
_session: &mut SessionState,
_node_id: usize
) -> Result<Option<Box<dyn OpState + 'static>>, TractError>
[src]
&self,
_session: &mut SessionState,
_node_id: usize
) -> Result<Option<Box<dyn OpState + 'static>>, TractError>
fn as_stateless(&self) -> Option<&(dyn StatelessOp + 'static)>
[src]
impl<T> ToOwned for T where
T: Clone,
[src]
T: Clone,
type Owned = T
The resulting type after obtaining ownership.
fn to_owned(&self) -> T
[src]
fn clone_into(&self, target: &mut T)
[src]
impl<T, U> TryFrom<U> for T where
U: Into<T>,
[src]
U: Into<T>,
type Error = Infallible
The type returned in the event of a conversion error.
fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>
[src]
impl<T, U> TryInto<U> for T where
U: TryFrom<T>,
[src]
U: TryFrom<T>,