Struct tract_core::ops::array::StridedSlice

pub struct StridedSlice {
    pub optional_axes_input: Option<usize>,
    pub optional_steps_input: Option<usize>,
    pub begin_mask: i64,
    pub end_mask: i64,
    pub shrink_axis_mask: i64,
}

Fields

optional_axes_input: Option<usize>

optional_steps_input: Option<usize>

begin_mask: i64

end_mask: i64

shrink_axis_mask: i64

Methods

impl StridedSlice

pub fn tensorflow(
    begin_mask: i64,
    end_mask: i64,
    shrink_axis_mask: i64
) -> StridedSlice

pub fn onnx10(
    optional_axes_input: Option<usize>,
    optional_steps_input: Option<usize>
) -> StridedSlice

Trait Implementations

impl InferenceRulesOp for StridedSlice

fn rules<'r, 'p: 'r, 's: 'r>(
    &'s self,
    s: &mut Solver<'r>,
    inputs: &'p [TensorProxy],
    outputs: &'p [TensorProxy]
) -> InferenceResult

Registers the inference rules of the operator.

fn as_op(&self) -> &dyn Op

fn as_op_mut(&mut self) -> &mut dyn Op

fn to_typed(
    &self,
    _source: &InferenceModel,
    node: &InferenceNode,
    target: &mut TypedModel,
    mapping: &HashMap<OutletId, OutletId>
) -> TractResult<TVec<OutletId>>

fn nboutputs(&self) -> TractResult<usize>

impl StatelessOp for StridedSlice

fn eval(&self, inputs: TVec<Arc<Tensor>>) -> TractResult<TVec<Arc<Tensor>>>

impl Op for StridedSlice

fn name(&self) -> Cow<str>

fn as_typed(&self) -> Option<&dyn TypedOp>

fn incorporate(
    &self,
    _model: &InferenceModel,
    _node: &InferenceNode
) -> TractResult<Option<InferenceModelPatch>>

Early pass on inference model, after analyse, but before translation to typed network. Meant to deal with some framework idiosyncrasies that manifest with temporaries nodes that can run some form of inference but require refactoring the network before it can be evaluated.

fn fuse(
    &self,
    _model: &TypedModel,
    _node: &TypedNode
) -> TractResult<Option<TypedModelPatch>>

Fuse op after codegen to deal with local optimisations.

fn nested_models(&self) -> Vec<(Cow<str>, &dyn Model)>

Nested models, with label (for audit).

fn validation(&self) -> Validation

The kind of accuracy check that should be performed on operation when testing them.

fn same_as(&self, _other: &dyn Op) -> bool

Compare two ops.

fn info(&self) -> TractResult<Vec<String>>

Short (one-line) strings giving hints on internal implementation or important configuration details to be displayed in dumps.

fn as_pulsed(&self) -> Option<&dyn PulsedOp>

fn is_canonic(&self) -> bool

impl Clone for StridedSlice

fn clone(&self) -> StridedSlice

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for StridedSlice

fn fmt(&self, f: &mut Formatter) -> Result

Formats the value using the given formatter.