From 636ab40d3741e12eaad11d5b50e4b34bfbb258b5 Mon Sep 17 00:00:00 2001 From: Aron Virginas-Tar Date: Mon, 16 Sep 2019 14:27:45 +0100 Subject: IVGCVSW-3875 Add frontend for SLICE layer Signed-off-by: Aron Virginas-Tar Change-Id: Iebe675a0cee02db6f133d48ce58cbc1e233061db --- include/armnn/Descriptors.hpp | 20 +++++++++++++++++++- 1 file changed, 19 insertions(+), 1 deletion(-) (limited to 'include/armnn/Descriptors.hpp') 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& begin, const std::vector& size) + : m_Begin(begin) + , m_Size(size) + {} + + SliceDescriptor() : SliceDescriptor({}, {}) + {} + + /// Beginning indices of the slice in each dimension. + std::vector m_Begin; + + /// Size of the slice in each dimension. + std::vector 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 -- cgit v1.2.1