diff options
Diffstat (limited to 'include/armnn/Descriptors.hpp')
-rw-r--r-- | include/armnn/Descriptors.hpp | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp index 5a7a647ee7..656afb1756 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -333,6 +333,24 @@ struct ReshapeDescriptor TensorShape m_TargetShape; }; +struct SpaceToBatchNdDescriptor +{ + SpaceToBatchNdDescriptor() + : m_DataLayout(DataLayout::NCHW) + {} + + SpaceToBatchNdDescriptor(const std::vector<unsigned int>& blockShape, + const std::vector<std::pair<unsigned int, unsigned int>>& padList) + : m_BlockShape(blockShape) + , m_PadList(padList) + , m_DataLayout(DataLayout::NCHW) + {} + + std::vector<unsigned int> m_BlockShape; + std::vector<std::pair<unsigned int, unsigned int>> m_PadList; + DataLayoutIndexed m_DataLayout; +}; + // temporary descriptor for Lstm struct LstmDescriptor { |