ArmNN
 24.02
StridedSliceLayer.cpp
Go to the documentation of this file.
1 //
2 // Copyright © 2018-2023 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 #include "StridedSliceLayer.hpp"
6 
7 #include "LayerCloneBase.hpp"
8 
10 
13 
14 namespace armnn
15 {
16 
18  : LayerWithParameters(1, 1, LayerType::StridedSlice, param, name)
19 {
20 }
21 
22 std::unique_ptr<IWorkload> StridedSliceLayer::CreateWorkload(const IWorkloadFactory& factory) const
23 {
24  StridedSliceQueueDescriptor descriptor;
25 
26  descriptor.m_Parameters.m_Begin = m_Param.m_Begin;
27  descriptor.m_Parameters.m_End = m_Param.m_End;
29 
30  // Optional parameters
36 
37  SetAdditionalInfo(descriptor);
38 
39  return factory.CreateWorkload(LayerType::StridedSlice, descriptor, PrepInfoAndDesc(descriptor));
40 }
41 
43 {
44  return CloneBase<StridedSliceLayer>(graph, m_Param, GetName());
45 }
46 
47 std::vector<TensorShape> StridedSliceLayer::InferOutputShapes(
48  const std::vector<TensorShape>& inputShapes) const
49 {
50  ARMNN_ASSERT(inputShapes.size() == 1);
51 
52  TensorShape inputShape = inputShapes[0];
53  std::vector<unsigned int> outputShape;
54  unsigned int amountDimShrunk{0};
55 
56  for (unsigned int i = 0; i < inputShape.GetNumDimensions(); i++)
57  {
58  int stride = m_Param.m_Stride[i];
59  int start = m_Param.GetStartForAxis(inputShape, i);
60  int stop = m_Param.GetStopForAxis(inputShape, i, start);
61 
62  if (m_Param.m_ShrinkAxisMask & (1 << i))
63  {
64  amountDimShrunk+=1;
65 
66  // If the difference between the start point and the end point of the slice on an axis being shrunk
67  // is greater than 1 then throw an error as the output will not be large enough to hold the slice
68  if (((m_Param.m_Begin[i] - m_Param.m_End[i]) > 1) || ((m_Param.m_Begin[i] - m_Param.m_End[i]) < -1))
69  {
71  "StridedSlice: Attempting to take a larger slice than can fit in inferred output");
72  }
73 
74  if (stride < 0)
75  {
77  "StridedSlice: Stride can not be negative with Shrink Axis Mask set.");
78  }
79  continue;
80  }
81 
82  int newSize = stride > 0 ? ((stop - start) + stride - 1) / stride :
83  ((start - stop) - stride - 1) / -stride;
84 
85  newSize = std::max(0, newSize);
86 
87  outputShape.push_back(armnn::numeric_cast<unsigned int>(newSize));
88  }
89 
90  if (outputShape.size() == 0 && (inputShape.GetNumDimensions() - amountDimShrunk) == 0)
91  {
92  outputShape.push_back(1);
93  }
94 
95  return std::vector<TensorShape>({
96  TensorShape(armnn::numeric_cast<unsigned int>(outputShape.size()), &outputShape[0]) });
97 }
98 
100 {
102 
103  const TensorShape& outputShape = GetOutputSlot(0).GetTensorInfo().GetShape();
104 
106 
107  auto inferredShapes = InferOutputShapes({GetInputSlot(0).GetTensorInfo().GetShape()});
108 
109  ARMNN_ASSERT(inferredShapes.size() == 1);
110 
111  ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "StridedSliceLayer");
112 }
113 
115 {
116  strategy.ExecuteStrategy(this, GetParameters(), {}, GetName());
117 }
118 
119 } // namespace armnn
ARMNN_ASSERT
#define ARMNN_ASSERT(COND)
Definition: Assert.hpp:14
armnn::OutputSlot::GetTensorInfo
const TensorInfo & GetTensorInfo() const override
Definition: Layer.cpp:92
WorkloadData.hpp
armnn::StridedSliceDescriptor::m_Begin
std::vector< int > m_Begin
Begin values for the input that will be sliced.
Definition: Descriptors.hpp:1342
armnn::StridedSliceDescriptor::m_EllipsisMask
int32_t m_EllipsisMask
Ellipsis mask value.
Definition: Descriptors.hpp:1357
CHECK_LOCATION
#define CHECK_LOCATION()
Definition: Exceptions.hpp:203
armnn::LayerType::StridedSlice
@ StridedSlice
armnn::Layer::ValidateAndCopyShape
void ValidateAndCopyShape(const TensorShape &outputShape, const TensorShape &inferredShape, const ShapeInferenceMethod shapeInferenceMethod, const std::string &layerName, const unsigned int outputSlotIndex=0)
Definition: Layer.cpp:435
armnn::Layer::GetOutputSlot
const OutputSlot & GetOutputSlot(unsigned int index=0) const override
Get the const output slot handle by slot index.
Definition: Layer.hpp:339
armnn::StridedSliceDescriptor::m_BeginMask
int32_t m_BeginMask
Begin mask value.
Definition: Descriptors.hpp:1350
armnn::IStrategy
Definition: IStrategy.hpp:16
armnn::Layer::GetInputSlot
const InputSlot & GetInputSlot(unsigned int index) const override
Get a const input slot handle by slot index.
Definition: Layer.hpp:337
armnn::LayerWithParameters< StridedSliceDescriptor >::GetParameters
const StridedSliceDescriptor & GetParameters() const override
Definition: LayerWithParameters.hpp:19
WorkloadFactory.hpp
NumericCast.hpp
armnn::LayerWithParameters
Definition: LayerWithParameters.hpp:14
armnn::StridedSliceLayer::Clone
StridedSliceLayer * Clone(Graph &graph) const override
Creates a dynamically-allocated copy of this layer.
Definition: StridedSliceLayer.cpp:42
armnn::Layer::GetName
const char * GetName() const override
Returns the name of the layer.
Definition: Layer.hpp:332
armnn::StridedSliceDescriptor::GetStopForAxis
int GetStopForAxis(const TensorShape &inputShape, unsigned int axis, int startForAxis) const
Definition: Descriptors.cpp:419
armnn::InputSlot::GetTensorInfo
const TensorInfo & GetTensorInfo() const override
Gets the TensorInfo for this InputSlot.
Definition: Layer.cpp:592
armnn::TensorShape
Definition: Tensor.hpp:20
armnn::LayerWithParameters< StridedSliceDescriptor >::m_Param
StridedSliceDescriptor m_Param
The parameters for the layer (not including tensor-valued weights etc.).
Definition: LayerWithParameters.hpp:52
armnn::QueueDescriptorWithParameters::m_Parameters
LayerDescriptor m_Parameters
Definition: WorkloadData.hpp:66
armnn::TensorShape::GetNumDimensions
unsigned int GetNumDimensions() const
Function that returns the tensor rank.
Definition: Tensor.cpp:174
armnn::StridedSliceLayer::ValidateTensorShapesFromInputs
void ValidateTensorShapesFromInputs() override
Check if the input tensor shape(s) will lead to a valid configuration of StridedSliceLayer.
Definition: StridedSliceLayer.cpp:99
armnn::StridedSliceQueueDescriptor
Definition: WorkloadData.hpp:467
armnn::LayerWithParameters< StridedSliceDescriptor >::PrepInfoAndDesc
WorkloadInfo PrepInfoAndDesc(QueueDescriptor &descriptor) const
Helper function to reduce duplication in *Layer::CreateWorkload.
Definition: LayerWithParameters.hpp:44
armnn::LayerValidationException
Definition: Exceptions.hpp:105
armnn::IWorkloadFactory
Definition: WorkloadFactory.hpp:22
armnn::Layer::VerifyShapeInferenceType
void VerifyShapeInferenceType(const TensorShape &outputShape, ShapeInferenceMethod shapeInferenceMethod)
Definition: Layer.cpp:504
armnn::Layer::SetAdditionalInfo
void SetAdditionalInfo(QueueDescriptor &descriptor) const
Definition: Layer.cpp:287
armnn::StridedSliceDescriptor::m_EndMask
int32_t m_EndMask
End mask value.
Definition: Descriptors.hpp:1353
armnn::StridedSliceDescriptor::m_ShrinkAxisMask
int32_t m_ShrinkAxisMask
Shrink axis mask value. If set, the nth specification shrinks the dimensionality by 1.
Definition: Descriptors.hpp:1355
armnn::StridedSliceLayer
This layer represents a strided slice operation.
Definition: StridedSliceLayer.hpp:13
armnn::StridedSliceDescriptor::m_Stride
std::vector< int > m_Stride
Stride values for the input that will be sliced.
Definition: Descriptors.hpp:1346
armnn::StridedSliceDescriptor
A StridedSliceDescriptor for the StridedSliceLayer.
Definition: Descriptors.hpp:1303
armnn::StridedSliceLayer::CreateWorkload
virtual std::unique_ptr< IWorkload > CreateWorkload(const IWorkloadFactory &factory) const override
Makes a workload for the StridedSlice type.
Definition: StridedSliceLayer.cpp:22
armnn::TensorInfo::GetShape
const TensorShape & GetShape() const
Definition: Tensor.hpp:193
armnn::StridedSliceDescriptor::GetStartForAxis
int GetStartForAxis(const TensorShape &inputShape, unsigned int axis) const
Definition: Descriptors.cpp:392
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6
armnn::StridedSlice
void StridedSlice(const TensorInfo &inputInfo, const StridedSliceDescriptor &params, const void *inputData, void *outputData, unsigned int dataTypeSize)
Definition: StridedSlice.cpp:90
armnn::StridedSliceDescriptor::m_End
std::vector< int > m_End
End values for the input that will be sliced.
Definition: Descriptors.hpp:1344
StridedSliceLayer.hpp
armnn::Layer::VerifyLayerConnections
void VerifyLayerConnections(unsigned int expectedConnections, const CheckLocation &location) const
Definition: Layer.cpp:391
armnn::StridedSliceLayer::StridedSliceLayer
StridedSliceLayer(const StridedSliceDescriptor &param, const char *name)
Constructor to create a StridedSliceLayer.
Definition: StridedSliceLayer.cpp:17
armnn::StridedSliceLayer::InferOutputShapes
std::vector< TensorShape > InferOutputShapes(const std::vector< TensorShape > &inputShapes) const override
By default returns inputShapes if the number of inputs are equal to number of outputs,...
Definition: StridedSliceLayer.cpp:47
armnn::StridedSliceLayer::ExecuteStrategy
void ExecuteStrategy(IStrategy &strategy) const override
Apply a visitor to this layer.
Definition: StridedSliceLayer.cpp:114
armnn::Layer::m_ShapeInferenceMethod
ShapeInferenceMethod m_ShapeInferenceMethod
Definition: Layer.hpp:441
armnn::LayerType
LayerType
When adding a new layer, adapt also the LastLayer enum value in the enum class LayerType below.
Definition: Types.hpp:491
armnn::StridedSliceDescriptor::m_NewAxisMask
int32_t m_NewAxisMask
New axis mask value.
Definition: Descriptors.hpp:1360
armnn::Graph
Definition: Graph.hpp:30
armnn::IWorkloadFactory::CreateWorkload
virtual std::unique_ptr< IWorkload > CreateWorkload(LayerType type, const QueueDescriptor &descriptor, const WorkloadInfo &info) const =0
Backends should implement their own CreateWorkload function with a switch statement.
armnn::IStrategy::ExecuteStrategy
virtual void ExecuteStrategy(const IConnectableLayer *layer, const armnn::BaseDescriptor &descriptor, const std::vector< armnn::ConstTensor > &constants, const char *name, const armnn::LayerBindingId id=0)=0
LayerCloneBase.hpp