diff options
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 |