aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/backends/WorkloadData.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/backends/WorkloadData.hpp')
-rw-r--r--src/armnn/backends/WorkloadData.hpp9
1 files changed, 9 insertions, 0 deletions
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;
};