1 files changed, 23 insertions, 0 deletions
diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp
index cb76615889..377f0705d7 100644
--- a/include/armnn/Descriptors.hpp
+++ b/include/armnn/Descriptors.hpp
@@ -648,6 +648,29 @@ struct PadDescriptor
     float m_PadValue;
 };
 
+/// A StackDescriptor for the StackLayer.
+struct StackDescriptor
+{
+    StackDescriptor()
+    : m_Axis(0)
+    , m_NumInputs(0)
+    , m_InputShape()
+    {}
+
+    StackDescriptor(uint32_t axis, uint32_t numInputs, const TensorShape& inputShape)
+    : m_Axis(axis)
+    , m_NumInputs(numInputs)
+    , m_InputShape(inputShape)
+    {}
+
+    /// 0-based axis along which to stack the input tensors.
+    uint32_t m_Axis;
+    /// Number of input tensors.
+    uint32_t m_NumInputs;
+    /// Required shape of all input tensors.
+    TensorShape m_InputShape;
+};
+
 /// A StridedSliceDescriptor for the StridedSliceLayer.
 struct StridedSliceDescriptor
 {