From 430b5d824a17ddfb31bb29c18407e45e0def958c Mon Sep 17 00:00:00 2001 From: Conor Kennedy Date: Wed, 14 Nov 2018 15:28:28 +0000 Subject: IVGCVSW-2086: Add StridedSlice layer & corresponding no-op factory implementations Change-Id: I906dcb8b4c9b491bf3a661f208c09d0ea1c0fa35 --- include/armnn/Descriptors.hpp | 33 +++++++++++++++++++++++++++++++++ 1 file changed, 33 insertions(+) (limited to 'include/armnn/Descriptors.hpp') diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp index 396e427ee4..f8dd18ab8a 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -421,4 +421,37 @@ struct PadDescriptor std::vector> m_PadList; }; +struct StridedSliceDescriptor +{ + StridedSliceDescriptor() + : m_DataLayout(DataLayout::NCHW) + {} + + StridedSliceDescriptor(const std::vector& begin, + const std::vector& end, + const std::vector& stride) + : m_Begin(begin) + , m_End(end) + , m_Stride(stride) + , m_BeginMask(0) + , m_EndMask(0) + , m_ShrinkAxisMask(0) + , m_EllipsisMask(0) + , m_NewAxisMask(0) + , m_DataLayout(DataLayout::NCHW) + {} + + std::vector m_Begin; + std::vector m_End; + std::vector m_Stride; + + int32_t m_BeginMask; + int32_t m_EndMask; + int32_t m_ShrinkAxisMask; + int32_t m_EllipsisMask; + int32_t m_NewAxisMask; + + DataLayoutIndexed m_DataLayout; +}; + } -- cgit v1.2.1