diff options
| author | Sadik Armagan <sadik.armagan@arm.com> | 2020-06-04 10:32:18 +0100 |
|---|---|---|
| committer | Sadik Armagan <sadik.armagan@arm.com> | 2020-06-05 09:48:47 +0000 |
| commit | 041b3c0d416d94b1aa4efa41960808b7e5e61f25 (patch) | |
| tree | 029b7f3b072d94df8cc92e92ab0a0300130c7902 /src/backends/reference/workloads/RefPadWorkload.cpp | |
| parent | fa0d838fd7be90b6e1d53aa62da14ccc0cf0eacc (diff) | |
| download | armnn-041b3c0d416d94b1aa4efa41960808b7e5e61f25.tar.gz | |
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
Diffstat (limited to 'src/backends/reference/workloads/RefPadWorkload.cpp')
| -rw-r--r-- | src/backends/reference/workloads/RefPadWorkload.cpp | 28 |
1 files changed, 6 insertions, 22 deletions
diff --git a/src/backends/reference/workloads/RefPadWorkload.cpp b/src/backends/reference/workloads/RefPadWorkload.cpp index 6f82d5ffdb..af22c31001 100644 --- a/src/backends/reference/workloads/RefPadWorkload.cpp +++ b/src/backends/reference/workloads/RefPadWorkload.cpp @@ -6,38 +6,22 @@ #include "RefPadWorkload.hpp" #include "Pad.hpp" -#include "RefWorkloadUtils.hpp" - #include "Profiling.hpp" - -#include <ResolveType.hpp> - -#include <vector> +#include "RefWorkloadUtils.hpp" namespace armnn { -template <armnn::DataType DataType> -void RefPadWorkload<DataType>::Execute() const +void RefPadWorkload::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& inputInfo = GetTensorInfo(m_Data.m_Inputs[0]); const TensorInfo& outputInfo = GetTensorInfo(m_Data.m_Outputs[0]); - 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, m_Data.m_Parameters.m_PadValue); + armnn::Pad(inputInfo, + outputInfo, + m_Data); } -template class RefPadWorkload<DataType::BFloat16>; -template class RefPadWorkload<DataType::Float32>; -template class RefPadWorkload<DataType::Float16>; -template class RefPadWorkload<DataType::QAsymmS8>; -template class RefPadWorkload<DataType::QAsymmU8>; -template class RefPadWorkload<DataType::QSymmS16>; - } //namespace armnn
\ No newline at end of file |