aboutsummaryrefslogtreecommitdiff
path: root/include/armnn/Descriptors.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'include/armnn/Descriptors.hpp')
-rw-r--r--include/armnn/Descriptors.hpp20
1 files changed, 19 insertions, 1 deletions
diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp
index e871e89ebb..8d382f700e 100644
--- a/include/armnn/Descriptors.hpp
+++ b/include/armnn/Descriptors.hpp
@@ -667,6 +667,24 @@ struct PadDescriptor
float m_PadValue;
};
+/// A SliceDescriptor for the SliceLayer.
+struct SliceDescriptor
+{
+ SliceDescriptor(const std::vector<unsigned int>& begin, const std::vector<unsigned int>& size)
+ : m_Begin(begin)
+ , m_Size(size)
+ {}
+
+ SliceDescriptor() : SliceDescriptor({}, {})
+ {}
+
+ /// Beginning indices of the slice in each dimension.
+ std::vector<unsigned int> m_Begin;
+
+ /// Size of the slice in each dimension.
+ std::vector<unsigned int> m_Size;
+};
+
/// A StackDescriptor for the StackLayer.
struct StackDescriptor
{
@@ -786,4 +804,4 @@ struct TransposeConvolution2dDescriptor
DataLayout m_DataLayout;
};
-} // namespace armnn \ No newline at end of file
+} // namespace armnn