From 041b3c0d416d94b1aa4efa41960808b7e5e61f25 Mon Sep 17 00:00:00 2001
From: Sadik Armagan <sadik.armagan@arm.com>
Date: Thu, 4 Jun 2020 10:32:18 +0100
Subject: IVGCVSW-4904 Refactor CpuRef PAD Workload

* Refactored templated workload creation
* Added int8_t unit tests

Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I018b7f9f20496b5f9d7445901fe0d3dd04199cd0
---
 .../reference/workloads/RefPadWorkload.hpp         | 25 +++-------------------
 1 file changed, 3 insertions(+), 22 deletions(-)

(limited to 'src/backends/reference/workloads/RefPadWorkload.hpp')

diff --git a/src/backends/reference/workloads/RefPadWorkload.hpp b/src/backends/reference/workloads/RefPadWorkload.hpp
index 74dcab1967..0b8379a60f 100644
--- a/src/backends/reference/workloads/RefPadWorkload.hpp
+++ b/src/backends/reference/workloads/RefPadWorkload.hpp
@@ -8,33 +8,14 @@
 #include <backendsCommon/Workload.hpp>
 #include <backendsCommon/WorkloadData.hpp>
 
-#include <armnn/TypesUtils.hpp>
-
 namespace armnn
 {
 
-template <armnn::DataType DataType>
-class RefPadWorkload : public TypedWorkload<PadQueueDescriptor, DataType>
+class RefPadWorkload : public BaseWorkload<PadQueueDescriptor>
 {
 public:
-
-    static const std::string& GetName()
-    {
-        static const std::string name = std::string("RefPad") + GetDataTypeName(DataType) + "Workload";
-        return name;
-    }
-
-    using TypedWorkload<PadQueueDescriptor, DataType>::m_Data;
-    using TypedWorkload<PadQueueDescriptor, DataType>::TypedWorkload;
-
-    void Execute() const override;
+    using BaseWorkload<PadQueueDescriptor>::BaseWorkload;
+    virtual void Execute() const override;
 };
 
-using RefPadBFloat16Workload = RefPadWorkload<DataType::BFloat16>;
-using RefPadFloat32Workload =  RefPadWorkload<DataType::Float32>;
-using RefPadFloat16Workload =  RefPadWorkload<DataType::Float16>;
-using RefPadQAsymmS8Workload = RefPadWorkload<DataType::QAsymmS8>;
-using RefPadQAsymm8Workload =  RefPadWorkload<DataType::QAsymmU8>;
-using RefPadQSymm16Workload =  RefPadWorkload<DataType::QSymmS16>;
-
 } //namespace armnn
-- 
cgit v1.2.1