diff options
author | Mohamed Nour Abouelseoud <mohamednour.abouelseoud@arm.com> | 2018-10-18 12:26:19 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-22 16:57:54 +0100 |
commit | dd6aceaa884815e68ed69fca71de81babd3204da (patch) | |
tree | 83a0e636a84833e7fd3e78b808d269f80d01a0f8 /src/backends/reference/workloads/RefPadWorkload.hpp | |
parent | d134093a271b60e248942af9757e8236e8f41ac1 (diff) | |
download | armnn-dd6aceaa884815e68ed69fca71de81babd3204da.tar.gz |
IVGCVSW-2013 Add a UInt8 Reference Implementation for the PAD Operator
Change-Id: I41f3606198db1fda8d72aaf5169594ba9156eb38
Diffstat (limited to 'src/backends/reference/workloads/RefPadWorkload.hpp')
-rw-r--r-- | src/backends/reference/workloads/RefPadWorkload.hpp | 25 |
1 files changed, 20 insertions, 5 deletions
diff --git a/src/backends/reference/workloads/RefPadWorkload.hpp b/src/backends/reference/workloads/RefPadWorkload.hpp index 7ff117d6a5..938fcf2004 100644 --- a/src/backends/reference/workloads/RefPadWorkload.hpp +++ b/src/backends/reference/workloads/RefPadWorkload.hpp @@ -5,17 +5,32 @@ #pragma once -#include "backends/Workload.hpp" -#include "backends/WorkloadData.hpp" +#include <backends/Workload.hpp> +#include <backends/WorkloadData.hpp> + +#include <armnn/TypesUtils.hpp> namespace armnn { -class RefPadWorkload : public BaseWorkload<PadQueueDescriptor> +template <armnn::DataType DataType> +class RefPadWorkload : public TypedWorkload<PadQueueDescriptor, DataType> { public: - explicit RefPadWorkload (const PadQueueDescriptor& descriptor, const WorkloadInfo& info); - virtual void Execute() const override; + + 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 RefPadFloat32Workload = RefPadWorkload<DataType::Float32>; +using RefPadUint8Workload = RefPadWorkload<DataType::QuantisedAsymm8>; + } //namespace armnn |