diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-09-16 14:27:45 +0100 |
---|---|---|
committer | Áron Virginás-Tar <aron.virginas-tar@arm.com> | 2019-09-17 09:03:43 +0000 |
commit | 636ab40d3741e12eaad11d5b50e4b34bfbb258b5 (patch) | |
tree | defaba57dc28c7b5dbe19075e24b6c8c0cefc9b2 /include/armnn/Descriptors.hpp | |
parent | 4dc64a69ba383ece509d442598617445a3b4847f (diff) | |
download | armnn-636ab40d3741e12eaad11d5b50e4b34bfbb258b5.tar.gz |
IVGCVSW-3875 Add frontend for SLICE layer
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: Iebe675a0cee02db6f133d48ce58cbc1e233061db
Diffstat (limited to 'include/armnn/Descriptors.hpp')
-rw-r--r-- | include/armnn/Descriptors.hpp | 20 |
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 |