diff options
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 |