[−][src]Struct tract_core::ops::array::Gather
Fields
axis: usize
Implementations
impl Gather
[src]
impl Gather
[src]
pub fn compute_output_shape<D: DimLike>(
&self,
input_shape: &[D],
indices_shape: &[D]
) -> TractResult<TVec<D>>
[src]
&self,
input_shape: &[D],
indices_shape: &[D]
) -> TractResult<TVec<D>>
Trait Implementations
impl Clone for Gather
[src]
impl Debug for Gather
[src]
impl DynHash for Gather
[src]
impl EvalOp for Gather
[src]
pub fn is_stateless(&self) -> bool
[src]
pub fn eval(&self, inputs: TVec<Arc<Tensor>>) -> TractResult<TVec<Arc<Tensor>>>
[src]
pub fn state(
&self,
session: &mut SessionState,
node_id: usize
) -> TractResult<Option<Box<dyn OpState>>>
[src]
&self,
session: &mut SessionState,
node_id: usize
) -> TractResult<Option<Box<dyn OpState>>>
impl Hash for Gather
[src]
pub fn hash<__H: Hasher>(&self, state: &mut __H)
[src]
pub fn hash_slice<H>(data: &[Self], state: &mut H) where
H: Hasher,
1.3.0[src]
H: Hasher,
impl Op for Gather
[src]
pub fn name(&self) -> Cow<'_, str>
[src]
pub fn op_families(&self) -> &'static [&'static str]ⓘ
[src]
pub fn as_typed(&self) -> Option<&dyn TypedOp>
[src]
pub fn validation(&self) -> Validation
[src]
pub fn same_as(&self, _other: &dyn Op) -> bool
[src]
pub fn info(&self) -> TractResult<Vec<String>>
[src]
impl TypedOp for Gather
[src]
pub fn as_op(&self) -> &dyn Op
[src]
pub fn as_op_mut(&mut self) -> &mut dyn Op
[src]
pub fn output_facts(
&self,
inputs: &[&TypedFact]
) -> TractResult<TVec<TypedFact>>
[src]
&self,
inputs: &[&TypedFact]
) -> TractResult<TVec<TypedFact>>
pub fn declutter(
&self,
model: &TypedModel,
node: &TypedNode
) -> TractResult<Option<TypedModelPatch>>
[src]
&self,
model: &TypedModel,
node: &TypedNode
) -> TractResult<Option<TypedModelPatch>>
pub fn invariants(
&self,
model: &TypedModel,
node: &TypedNode
) -> TractResult<Invariants>
[src]
&self,
model: &TypedModel,
node: &TypedNode
) -> TractResult<Invariants>
pub fn fuse(
&self,
_model: &TypedModel,
_node: &TypedNode
) -> TractResult<Option<TypedModelPatch>>
[src]
&self,
_model: &TypedModel,
_node: &TypedNode
) -> TractResult<Option<TypedModelPatch>>
pub fn cost(&self, _inputs: &[&TypedFact]) -> TractResult<TVec<(Cost, TDim)>>
[src]
pub fn suggested_axis_changes(&self) -> TractResult<TVec<(InOut, AxisOp)>>
[src]
pub fn change_axes(
&self,
model: &TypedModel,
node: &TypedNode,
io: InOut,
change: &AxisOp
) -> TractResult<Option<AxisChangeConsequence>>
[src]
&self,
model: &TypedModel,
node: &TypedNode,
io: InOut,
change: &AxisOp
) -> TractResult<Option<AxisChangeConsequence>>
pub fn slice_output(
&self,
model: &TypedModel,
node: &TypedNode,
patch: &mut TypedModelPatch,
output_slot: usize,
axis: usize,
start: usize,
end: usize
) -> TractResult<Option<OutletId>>
[src]
&self,
model: &TypedModel,
node: &TypedNode,
patch: &mut TypedModelPatch,
output_slot: usize,
axis: usize,
start: usize,
end: usize
) -> TractResult<Option<OutletId>>
pub fn quantize(
&self,
model: &TypedModel,
node: &TypedNode,
dt: DatumType,
scale: f32,
zero_point: i32
) -> TractResult<Option<Box<dyn TypedOp>>>
[src]
&self,
model: &TypedModel,
node: &TypedNode,
dt: DatumType,
scale: f32,
zero_point: i32
) -> TractResult<Option<Box<dyn TypedOp>>>
pub fn concretize_dims(
&self,
source: &TypedModel,
node: &TypedNode,
target: &mut TypedModel,
mapping: &HashMap<OutletId, OutletId>,
values: &SymbolValues
) -> TractResult<TVec<OutletId>>
[src]
&self,
source: &TypedModel,
node: &TypedNode,
target: &mut TypedModel,
mapping: &HashMap<OutletId, OutletId>,
values: &SymbolValues
) -> TractResult<TVec<OutletId>>
pub fn codegen(
&self,
model: &TypedModel,
node: &TypedNode
) -> TractResult<Option<TypedModelPatch>>
[src]
&self,
model: &TypedModel,
node: &TypedNode
) -> TractResult<Option<TypedModelPatch>>
pub 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,
pub fn borrow_mut(&mut self) -> &mut T
[src]
impl<T> Downcast for T where
T: Any,
[src]
T: Any,
pub fn into_any(self: Box<T, Global>) -> Box<dyn Any + 'static, Global>
[src]
pub fn into_any_rc(self: Rc<T>) -> Rc<dyn Any + 'static>
[src]
pub fn as_any(&self) -> &(dyn Any + 'static)
[src]
pub fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)
[src]
impl<T> DowncastSync for T where
T: Send + Sync + Any,
[src]
T: Send + Sync + Any,
impl<T> DynClone for T where
T: Clone,
[src]
T: Clone,
pub fn __clone_box(&self, Private) -> *mut ()
[src]
impl<T> From<T> for T
[src]
impl<T, U> Into<U> for T where
U: From<T>,
[src]
U: From<T>,
impl<T> ToOwned for T where
T: Clone,
[src]
T: Clone,
type Owned = T
The resulting type after obtaining ownership.
pub fn to_owned(&self) -> T
[src]
pub 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.
pub 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>,