aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/RefDebugWorkload.hpp
diff options
context:
space:
mode:
authorNattapat Chaimanowong <nattapat.chaimanowong@arm.com>2018-12-06 11:54:33 +0000
committerLes Bell <les.bell@arm.com>2018-12-07 08:50:53 +0000
commitcfdcadf9feebaa112545923343875e790a18cc1c (patch)
tree0255dc2008acaf43322a7965378a5a3c27b97f20 /src/backends/reference/workloads/RefDebugWorkload.hpp
parent23be07e855c066f192c1007d529064462853a27c (diff)
downloadarmnn-cfdcadf9feebaa112545923343875e790a18cc1c.tar.gz
IVGCVSW-2316 Add reference implementation and unit tests for Debug
Change-Id: Ib2e5de2a057da57ef77a9f5c4367d699d4773294
Diffstat (limited to 'src/backends/reference/workloads/RefDebugWorkload.hpp')
-rw-r--r--src/backends/reference/workloads/RefDebugWorkload.hpp33
1 files changed, 33 insertions, 0 deletions
diff --git a/src/backends/reference/workloads/RefDebugWorkload.hpp b/src/backends/reference/workloads/RefDebugWorkload.hpp
new file mode 100644
index 0000000000..a1231f92d3
--- /dev/null
+++ b/src/backends/reference/workloads/RefDebugWorkload.hpp
@@ -0,0 +1,33 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+#pragma once
+
+#include <backendsCommon/Workload.hpp>
+
+#include <armnn/TypesUtils.hpp>
+
+namespace armnn
+{
+
+template <armnn::DataType DataType>
+class RefDebugWorkload : public TypedWorkload<DebugQueueDescriptor, DataType>
+{
+public:
+ static const std::string& GetName()
+ {
+ static const std::string name = std::string("RefDebug") + GetDataTypeName(DataType) + "Workload";
+ return name;
+ }
+
+ using TypedWorkload<DebugQueueDescriptor, DataType>::m_Data;
+ using TypedWorkload<DebugQueueDescriptor, DataType>::TypedWorkload;
+
+ void Execute() const override;
+};
+
+using RefDebugFloat32Workload = RefDebugWorkload<DataType::Float32>;
+using RefDebugUint8Workload = RefDebugWorkload<DataType::QuantisedAsymm8>;
+
+} //namespace armnn