24.02.1
|
Go to the documentation of this file.
39 : _starts(starts), _ends(ends), _strides(strides), _info(std::move(
info))
75 info.begin_mask(),
info.end_mask(),
info.shrink_axis_mask());
static TensorDescriptor compute_output_descriptor(const TensorDescriptor &input_descriptor, const Coordinates &starts, const Coordinates &ends, const BiStrides &strides, StridedSliceLayerInfo info)
Computes slice layer output descriptor.
constexpr EdgeID EmptyEdgeID
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.
SimpleTensor< float > src
Coordinates ends() const
End coordinates accessor.
TensorID input_id(size_t idx) const
Returns the tensor ID of a given input of the node.
Tensor * output(size_t idx) const
Returns the tensor of a given output of the node.
#define ARM_COMPUTE_ERROR_ON(cond)
If the condition is true then an error message is printed and an exception thrown.
virtual void visit(INode &n)=0
Visit INode.
StridedSliceLayerInfo strided_slice_info() const
void accept(INodeVisitor &v) override
Accepts a node visitor.
NodeType type() const override
Returns node's type.
TensorShape shape
Tensor shape.
#define ARM_COMPUTE_UNUSED(...)
To avoid unused variables warnings.
Copyright (c) 2017-2024 Arm Limited.
Tensor * input(size_t idx) const
Returns the tensor of a given input of the node.
TensorID output_id(size_t idx) const
Returns the tensor ID of a given output of the node.
constexpr TensorID NullTensorID
Constant NodeID specifying an equivalent of null node.
StridedSliceLayerNode(const Coordinates &starts, const Coordinates &ends, const BiStrides &strides, StridedSliceLayerInfo strided_slice_info)
Default Constructor.
BiStrides strides() const
Strides vector accessor.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
Coordinates starts() const
Start coordinates accessor.
bool forward_descriptors() override
Forwards descriptor information to outputs if possible.