1 files changed, 14 insertions, 1 deletions
diff --git a/src/armnn/Layer.cpp b/src/armnn/Layer.cpp
index 48ab8b5dbe..b0a3ac543b 100644
--- a/src/armnn/Layer.cpp
+++ b/src/armnn/Layer.cpp
@@ -124,10 +124,15 @@ LayerGuid GenerateLayerGuid()
 }
 } // namespace
 
-Layer::Layer(unsigned int numInputSlots, unsigned int numOutputSlots, LayerType type, const char* name)
+Layer::Layer(unsigned int numInputSlots,
+             unsigned int numOutputSlots,
+             LayerType type,
+             DataLayout layout,
+             const char* name)
 : m_OutputHandlers(numOutputSlots)
 , m_LayerName(name ? name : "")
 , m_Type(type)
+, m_DataLayout(layout)
 , m_ComputeDevice(Compute::Undefined)
 , m_Guid(GenerateLayerGuid())
 {
@@ -144,6 +149,14 @@ Layer::Layer(unsigned int numInputSlots, unsigned int numOutputSlots, LayerType
     }
 }
 
+Layer::Layer(unsigned int numInputSlots,
+             unsigned int numOutputSlots,
+             LayerType type,
+             const char* name)
+: Layer(numInputSlots, numOutputSlots, type, DataLayout::NCHW, name)
+{
+}
+
 void Layer::CollectWorkloadInputs(WorkloadDataCollector& dataCollector, const Graph& graph) const
 {
     for (auto&& inputSlot : GetInputSlots())