[−][src]Struct tract_hir::ops::array::Gather
Implementations
impl Gather
[src]
pub fn compute_output_shape<D>(
&self,
input_shape: &[D],
indices_shape: &[D]
) -> Result<SmallVec<[D; 4]>, TractError> where
D: DimLike,
[src]
&self,
input_shape: &[D],
indices_shape: &[D]
) -> Result<SmallVec<[D; 4]>, TractError> where
D: DimLike,
impl Gather
[src]
Trait Implementations
impl Clone for Gather
[src]
impl Debug for Gather
[src]
impl DynHash for Gather
impl Hash for Gather
[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 InferenceRulesOp for Gather
[src]
fn rules<'r, 'p: 'r, 's: 'r>(
&'s self,
s: &mut Solver<'r>,
inputs: &'p [TensorProxy],
outputs: &'p [TensorProxy]
) -> InferenceResult
[src]
&'s self,
s: &mut Solver<'r>,
inputs: &'p [TensorProxy],
outputs: &'p [TensorProxy]
) -> InferenceResult
fn as_op(&self) -> &dyn Op
[src]
fn as_op_mut(&mut self) -> &mut dyn Op
[src]
fn to_typed(
&self,
_source: &InferenceModel,
node: &InferenceNode,
target: &mut TypedModel,
mapping: &HashMap<OutletId, OutletId>
) -> TractResult<TVec<OutletId>>
[src]
&self,
_source: &InferenceModel,
node: &InferenceNode,
target: &mut TypedModel,
mapping: &HashMap<OutletId, OutletId>
) -> TractResult<TVec<OutletId>>
fn nboutputs(&self) -> TractResult<usize>
[src]
fn incorporate(
&self,
model: &InferenceModel,
node: &InferenceNode
) -> TractResult<Option<InferenceModelPatch>>
[src]
&self,
model: &InferenceModel,
node: &InferenceNode
) -> TractResult<Option<InferenceModelPatch>>
impl Op for Gather
[src]
fn name(&self) -> Cow<str>
[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 nested_models(
&self
) -> Vec<(Cow<str>, &(dyn Model + 'static), Vec<String>, Vec<String>)>
[src]
&self
) -> Vec<(Cow<str>, &(dyn Model + 'static), Vec<String>, Vec<String>)>
fn validation(&self) -> Validation
[src]
fn same_as(&self, _other: &(dyn Op + 'static)) -> bool
[src]
fn info(&self) -> Result<Vec<String>, TractError>
[src]
fn is_canonic(&self) -> bool
[src]
impl StatelessOp for Gather
[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>
Evaluates the operation given the input tensors.
impl TypedOp for Gather
[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: &ModelImpl<TypedFact, Box<dyn TypedOp + 'static>>,
node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>
) -> Result<Invariants, TractError>
[src]
&self,
model: &ModelImpl<TypedFact, Box<dyn TypedOp + 'static>>,
node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>
) -> Result<Invariants, TractError>
fn fuse(
&self,
_model: &ModelImpl<TypedFact, Box<dyn TypedOp + 'static>>,
_node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>
) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp + 'static>>>, TractError>
[src]
&self,
_model: &ModelImpl<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: &ModelImpl<TypedFact, Box<dyn TypedOp + 'static>>,
node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>
) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp + 'static>>>, TractError>
[src]
&self,
model: &ModelImpl<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 suggested_axis_changes(
&self
) -> Result<SmallVec<[(InOut, AxisOp); 4]>, TractError>
[src]
&self
) -> Result<SmallVec<[(InOut, AxisOp); 4]>, TractError>
fn change_axes(
&self,
model: &ModelImpl<TypedFact, Box<dyn TypedOp + 'static>>,
node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>,
io: InOut,
change: &AxisOp
) -> Result<Option<AxisChangeConsequence>, TractError>
[src]
&self,
model: &ModelImpl<TypedFact, Box<dyn TypedOp + 'static>>,
node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>,
io: InOut,
change: &AxisOp
) -> Result<Option<AxisChangeConsequence>, TractError>
fn slice_output(
&self,
model: &ModelImpl<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: &ModelImpl<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: &ModelImpl<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: &ModelImpl<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 pulsify(
&self,
_source: &ModelImpl<NormalizedFact, Box<dyn TypedOp + 'static>>,
node: &BaseNode<NormalizedFact, Box<dyn TypedOp + 'static>>,
_target: &mut ModelImpl<PulsedFact, Box<dyn PulsedOp + 'static>>,
_mapping: &HashMap<OutletId, OutletId, RandomState>,
_pulse: usize
) -> Result<SmallVec<[OutletId; 4]>, TractError>
[src]
&self,
_source: &ModelImpl<NormalizedFact, Box<dyn TypedOp + 'static>>,
node: &BaseNode<NormalizedFact, Box<dyn TypedOp + 'static>>,
_target: &mut ModelImpl<PulsedFact, Box<dyn PulsedOp + 'static>>,
_mapping: &HashMap<OutletId, OutletId, RandomState>,
_pulse: usize
) -> Result<SmallVec<[OutletId; 4]>, TractError>
fn codegen(
&self,
_model: &ModelImpl<TypedFact, Box<dyn TypedOp + 'static>>,
_node: &BaseNode<TypedFact, Box<dyn TypedOp + 'static>>
) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp + 'static>>>, TractError>
[src]
&self,
_model: &ModelImpl<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 Gather
impl Send for Gather
impl Sync for Gather
impl Unpin for Gather
impl UnwindSafe for Gather
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,
impl<T> BorrowMut<T> for T where
T: ?Sized,
[src]
T: ?Sized,
fn borrow_mut(&mut self) -> &mut T
[src]
impl<T> Downcast for T where
T: Any,
T: Any,
fn into_any(self: Box<T>) -> Box<dyn Any + 'static>
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>,