diff options
Diffstat (limited to 'include')
| -rw-r--r-- | include/armnn/Descriptors.hpp | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp index b705abe729..22dd0d2991 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -427,10 +427,6 @@ struct PadDescriptor struct StridedSliceDescriptor { - StridedSliceDescriptor() - : m_DataLayout(DataLayout::NCHW) - {} - StridedSliceDescriptor(const std::vector<int>& begin, const std::vector<int>& end, const std::vector<int>& stride) @@ -445,6 +441,15 @@ struct StridedSliceDescriptor , m_DataLayout(DataLayout::NCHW) {} + StridedSliceDescriptor() + : StridedSliceDescriptor({}, {}, {}) + {} + + int GetStartForAxis(const TensorShape& inputShape, unsigned int axis) const; + int GetStopForAxis(const TensorShape& inputShape, + unsigned int axis, + int startForAxis) const; + std::vector<int> m_Begin; std::vector<int> m_End; std::vector<int> m_Stride; |