aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/RefPadWorkload.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/reference/workloads/RefPadWorkload.cpp')
-rw-r--r--src/backends/reference/workloads/RefPadWorkload.cpp28
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