IVGCVSW-1883 Add support for different memory layouts

Change-Id: I6e9973bf25acad980fb4e96af8080ac829db0d28

1 files changed, 4 insertions, 0 deletions

diff --git a/src/armnn/Layer.hpp b/src/armnn/Layer.hpp
index 31837a5cc0..fd523cee9e 100644
--- a/src/armnn/Layer.hpp
+++ b/src/armnn/Layer.hpp
@@ -186,6 +186,7 @@ class Layer : public IConnectableLayer
 public:
     /// @param name - Optional name for the layer (may be nullptr).
     Layer(unsigned int numInputSlots, unsigned int numOutputSlots, LayerType type, const char* name);
+    Layer(unsigned int numInputSlots, unsigned int numOutputSlots, LayerType type, DataLayout layout, const char* name);
 
     const std::string& GetNameStr() const
     {
@@ -234,6 +235,8 @@ public:
 
     DataType GetDataType() const;
 
+    DataLayout GetDataLayout() const { return m_DataLayout; }
+
     Compute GetComputeDevice() const { return m_ComputeDevice; }
     void SetComputeDevice(Compute device) { m_ComputeDevice = device; }
 
@@ -341,6 +344,7 @@ private:
     std::vector<OutputSlot> m_OutputSlots;
 
     const LayerType m_Type;
+    const DataLayout m_DataLayout;
     Compute m_ComputeDevice;
 
     /// Used for sorting.