diff options
author | narpra01 <narumol.prangnawarat@arm.com> | 2018-09-10 17:35:27 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-09-25 14:54:29 +0100 |
commit | eb06191bdaeab6640691a00349eb93141fb49fe8 (patch) | |
tree | b850e2eddee1e61f79a1b0efc6b6670a0920d88c /src/armnn/backends/WorkloadData.hpp | |
parent | 4a8692cf18ebd3c4de125274d5c840d7be64e3cd (diff) | |
download | armnn-eb06191bdaeab6640691a00349eb93141fb49fe8.tar.gz |
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
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; }; |