1 files changed, 12 insertions, 1 deletions
diff --git a/src/armnn/layers/LayerWithParameters.hpp b/src/armnn/layers/LayerWithParameters.hpp
index 0cb970a2f1..6156d6b66c 100644
--- a/src/armnn/layers/LayerWithParameters.hpp
+++ b/src/armnn/layers/LayerWithParameters.hpp
@@ -28,13 +28,24 @@ protected:
     LayerWithParameters(unsigned int numInputSlots,
                         unsigned int numOutputSlots,
                         LayerType type,
+                        DataLayout layout,
                         const Parameters& param,
                         const char* name)
-    :   Layer(numInputSlots, numOutputSlots, type, name)
+    :   Layer(numInputSlots, numOutputSlots, type, layout, name)
     ,   m_Param(param)
     {
     }
 
+    LayerWithParameters(unsigned int numInputSlots,
+                        unsigned int numOutputSlots,
+                        LayerType type,
+                        const Parameters& param,
+                        const char* name)
+        : Layer(numInputSlots, numOutputSlots, type, name)
+        , m_Param(param)
+    {
+    }
+
     ~LayerWithParameters() = default;
 
     /// Helper function to reduce duplication in *Layer::CreateWorkload.