1 files changed, 15 insertions, 1 deletions
diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp
index 25bf818987..9479db3571 100644
--- a/include/armnn/Descriptors.hpp
+++ b/include/armnn/Descriptors.hpp
@@ -540,6 +540,20 @@ struct SpaceToBatchNdDescriptor
     DataLayout m_DataLayout;
 };
 
+/// A SpaceToDepthDescriptor for the SpaceToDepthLayer
+struct SpaceToDepthDescriptor
+{
+    SpaceToDepthDescriptor()
+    : m_BlockSize(1u)
+    , m_DataLayout(DataLayout::NHWC)
+    {}
+
+    /// Scalar specifying the input block size. It must be >= 1
+    unsigned int m_BlockSize;
+    /// The data layout to be used (NCHW, NHWC).
+    DataLayout m_DataLayout;
+};
+
 /// An LstmDescriptor for the LstmLayer.
 struct LstmDescriptor
 {
@@ -667,4 +681,4 @@ struct PreCompiledDescriptor
     unsigned int m_NumOutputSlots;
 };
 
-}
+} // namespace armnn
+\ No newline at end of file