commit a05c2106ba5dd4361363049b0588d6995fd01f45
author Nikhil Raj <nikhil.raj@arm.com> Tue Sep 25 16:16:13 2018 +0100
committer Matthew Bentham <matthew.bentham@arm.com> Wed Oct 10 16:16:57 2018 +0100
tree 39b03ef4bcdd448df5a7ff76f56d7f7a8b0a42c9
parent 711fa31d5d43b904d28bcd407cd7e921529a37ca

IVGCVSW-1910 Add data layout parameter for DepthwiseConvolution

Change-Id: Ia14c9d8c0a38b669a7589e63d74424e398790e54

src/backends/WorkloadData.hpp

diff --git a/src/backends/WorkloadData.hpp b/src/backends/WorkloadData.hpp
index 5da9e8b..8c1cbd4 100644
--- a/src/backends/WorkloadData.hpp
+++ b/src/backends/WorkloadData.hpp
@@ -159,11 +159,13 @@
     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;
 };