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.cpp | |
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.cpp')
-rw-r--r-- | src/backends/reference/workloads/RefPadWorkload.cpp | 17 |
1 files changed, 10 insertions, 7 deletions
diff --git a/src/backends/reference/workloads/RefPadWorkload.cpp b/src/backends/reference/workloads/RefPadWorkload.cpp index 233fbe4f34..b41c2de9af 100644 --- a/src/backends/reference/workloads/RefPadWorkload.cpp +++ b/src/backends/reference/workloads/RefPadWorkload.cpp @@ -10,28 +10,31 @@ #include "Profiling.hpp" +#include "TypeUtils.hpp" + #include <vector> namespace armnn { -RefPadWorkload::RefPadWorkload(const PadQueueDescriptor& descriptor, const WorkloadInfo& info) - :BaseWorkload<PadQueueDescriptor>(descriptor, info) {} - - -void RefPadWorkload::Execute() const +template <armnn::DataType DataType> +void RefPadWorkload<DataType>::Execute() const { + using T = ResolveType<DataType>; ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefPadWorkload_Execute"); const TensorInfo& inputInfo = GetTensorInfo(m_Data.m_Inputs[0]); const TensorInfo& outputInfo = GetTensorInfo(m_Data.m_Outputs[0]); - const float* inputData = GetInputTensorDataFloat(0, m_Data); - float* outputData = GetOutputTensorDataFloat(0, m_Data); + const T* inputData = GetInputTensorData<T>(0, m_Data); + T* outputData = GetOutputTensorData<T>(0, m_Data); Pad(inputInfo, outputInfo, m_Data.m_Parameters.m_PadList, inputData, outputData); } +template class RefPadWorkload<DataType::Float32>; +template class RefPadWorkload<DataType::QuantisedAsymm8>; + } //namespace armnn
\ No newline at end of file |