From eb06191bdaeab6640691a00349eb93141fb49fe8 Mon Sep 17 00:00:00 2001
From: narpra01 <narumol.prangnawarat@arm.com>
Date: Mon, 10 Sep 2018 17:35:27 +0100
Subject: IVGCVSW-1831 - Add dimension check to MeanQueueDescriptor::Validate
 to check if the output dimension is correct from a given input and options.

Change-Id: Ibc15d9ea3151a7ba1935feafeb1843ee035e7f2e
---
 src/armnn/backends/WorkloadData.hpp | 9 +++++++++
 1 file changed, 9 insertions(+)

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

diff --git a/src/armnn/backends/WorkloadData.hpp b/src/armnn/backends/WorkloadData.hpp
index f8f7e326dc..face761e73 100644
--- a/src/armnn/backends/WorkloadData.hpp
+++ b/src/armnn/backends/WorkloadData.hpp
@@ -199,6 +199,15 @@ struct SubtractionQueueDescriptor : QueueDescriptor
 // Mean layer workload data.
 struct MeanQueueDescriptor : QueueDescriptor
 {
+    MeanQueueDescriptor()
+        : m_Axis(nullptr)
+        , m_Keepdims(false)
+    {
+    }
+
+    const ConstCpuTensorHandle* m_Axis;
+    bool m_Keepdims;
+
     void Validate(const WorkloadInfo& workloadInfo) const;
 };
 
-- 
cgit v1.2.1