aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/RefMeanFloat32Workload.hpp
diff options
context:
space:
mode:
authornarpra01 <narumol.prangnawarat@arm.com>2018-09-28 11:07:51 +0100
committerMatthew Bentham <matthew.bentham@arm.com>2018-10-10 16:16:57 +0100
commit1e4c31dafb1c8984a126fa1d211ed8f9eedaf7cc (patch)
tree006e40b3bbfdc4a202cdada8fa9afec0dd8fffae /src/backends/reference/workloads/RefMeanFloat32Workload.hpp
parent33cea4db0b2729c5dbd50f9c0985578c60baffdd (diff)
downloadarmnn-1e4c31dafb1c8984a126fa1d211ed8f9eedaf7cc.tar.gz
IVGCVSW-1812 Adding Ref implementation and tests of MeanWorkloads
Change-Id: I6fb15c407024e3b91d5abf4513f8090be5821760
Diffstat (limited to 'src/backends/reference/workloads/RefMeanFloat32Workload.hpp')
-rw-r--r--src/backends/reference/workloads/RefMeanFloat32Workload.hpp22
1 files changed, 22 insertions, 0 deletions
diff --git a/src/backends/reference/workloads/RefMeanFloat32Workload.hpp b/src/backends/reference/workloads/RefMeanFloat32Workload.hpp
new file mode 100644
index 0000000000..a4c559f0c6
--- /dev/null
+++ b/src/backends/reference/workloads/RefMeanFloat32Workload.hpp
@@ -0,0 +1,22 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#pragma once
+
+#include "backends/Workload.hpp"
+#include "backends/WorkloadData.hpp"
+
+namespace armnn
+{
+
+
+class RefMeanFloat32Workload : public Float32Workload<MeanQueueDescriptor>
+{
+public:
+ explicit RefMeanFloat32Workload (const MeanQueueDescriptor& descriptor, const WorkloadInfo& info);
+ virtual void Execute() const override;
+};
+
+}//namespace armnn