From a0d2844d05dd9ae15733d426b04ab651457604ce Mon Sep 17 00:00:00 2001
From: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Date: Wed, 21 Nov 2018 16:48:17 +0000
Subject: IVGCVSW-2086 Update StridedSliceLayer and StridedSliceDescriptor

Change-Id: Ifa88a879dd239f60ab27330d6b73859393828ef0
---
 include/armnn/Descriptors.hpp | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

(limited to 'include')

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;
-- 
cgit v1.2.1