1 files changed, 19 insertions, 0 deletions
diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp
index 656afb1756..bda8cf7396 100644
--- a/include/armnn/Descriptors.hpp
+++ b/include/armnn/Descriptors.hpp
@@ -296,6 +296,25 @@ struct BatchNormalizationDescriptor
     DataLayoutIndexed m_DataLayout;
 };
 
+struct BatchToSpaceNdDescriptor
+{
+    BatchToSpaceNdDescriptor()
+        : m_BlockShape()
+        , m_Crops()
+        , m_DataLayout(DataLayout::NCHW)
+    {}
+
+    BatchToSpaceNdDescriptor(std::vector<unsigned int> blockShape, std::vector<std::vector<unsigned int>> crops)
+        : m_BlockShape(blockShape)
+        , m_Crops(crops)
+        , m_DataLayout(DataLayout::NCHW)
+    {}
+
+    std::vector<unsigned int> m_BlockShape;
+    std::vector<std::vector<unsigned int>> m_Crops;
+    DataLayoutIndexed m_DataLayout;
+};
+
 struct FakeQuantizationDescriptor
 {
     FakeQuantizationDescriptor()