From a05c2106ba5dd4361363049b0588d6995fd01f45 Mon Sep 17 00:00:00 2001
From: Nikhil Raj <nikhil.raj@arm.com>
Date: Tue, 25 Sep 2018 16:16:13 +0100
Subject: IVGCVSW-1910 Add data layout parameter for DepthwiseConvolution

Change-Id: Ia14c9d8c0a38b669a7589e63d74424e398790e54
---
 src/backends/WorkloadData.hpp | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'src/backends/WorkloadData.hpp')

diff --git a/src/backends/WorkloadData.hpp b/src/backends/WorkloadData.hpp
index 5da9e8b1fd..8c1cbd4964 100644
--- a/src/backends/WorkloadData.hpp
+++ b/src/backends/WorkloadData.hpp
@@ -159,11 +159,13 @@ struct DepthwiseConvolution2dQueueDescriptor : QueueDescriptorWithParameters<Dep
     DepthwiseConvolution2dQueueDescriptor()
         : m_Weight(nullptr)
         , m_Bias(nullptr)
+        , m_DataLayout(DataLayout::NCHW)
     {
     }
 
     const ConstCpuTensorHandle* m_Weight;
     const ConstCpuTensorHandle* m_Bias;
+    DataLayout m_DataLayout;
 
     void Validate(const WorkloadInfo& workloadInfo) const;
 };
-- 
cgit v1.2.1