diff options
Diffstat (limited to 'src/armnn/backends/WorkloadData.hpp')
-rw-r--r-- | src/armnn/backends/WorkloadData.hpp | 9 |
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; }; |