diff options
| author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2018-11-21 16:48:17 +0000 |
|---|---|---|
| committer | nattapat.chaimanowong <nattapat.chaimanowong@arm.com> | 2018-11-21 17:13:30 +0000 |
| commit | a0d2844d05dd9ae15733d426b04ab651457604ce (patch) | |
| tree | 7ae0787d7be49ec2b1bb754ff35e13e8b71c73d2 /include | |
| parent | 3d93bc47f42b339d82cfcf56a90c9264dd46d70a (diff) | |
| download | armnn-a0d2844d05dd9ae15733d426b04ab651457604ce.tar.gz | |
IVGCVSW-2086 Update StridedSliceLayer and StridedSliceDescriptor
Change-Id: Ifa88a879dd239f60ab27330d6b73859393828ef0
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; |