aboutsummaryrefslogtreecommitdiff
path: root/src/backends/ClWorkloads/ClWorkloadUtils.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/ClWorkloads/ClWorkloadUtils.hpp')
-rw-r--r--src/backends/ClWorkloads/ClWorkloadUtils.hpp19
1 files changed, 7 insertions, 12 deletions
diff --git a/src/backends/ClWorkloads/ClWorkloadUtils.hpp b/src/backends/ClWorkloads/ClWorkloadUtils.hpp
index a10237cf40..9c415b2267 100644
--- a/src/backends/ClWorkloads/ClWorkloadUtils.hpp
+++ b/src/backends/ClWorkloads/ClWorkloadUtils.hpp
@@ -20,7 +20,7 @@ namespace armnn
{
template <typename T>
-void CopyArmComputeClTensorData(const T* srcData, arm_compute::CLTensor& dstTensor)
+void CopyArmComputeClTensorData(arm_compute::CLTensor& dstTensor, const T* srcData)
{
{
ARMNN_SCOPED_PROFILING_EVENT_CL("MapClTensorForWriting");
@@ -35,30 +35,25 @@ void CopyArmComputeClTensorData(const T* srcData, arm_compute::CLTensor& dstTens
dstTensor.unmap();
}
-template <typename T>
-void InitialiseArmComputeClTensorData(arm_compute::CLTensor& clTensor, const T* data)
-{
- armcomputetensorutils::InitialiseArmComputeTensorEmpty(clTensor);
- CopyArmComputeClTensorData<T>(data, clTensor);
-}
-
inline void InitializeArmComputeClTensorData(arm_compute::CLTensor& clTensor,
const ConstCpuTensorHandle* handle)
{
BOOST_ASSERT(handle);
+
+ armcomputetensorutils::InitialiseArmComputeTensorEmpty(clTensor);
switch(handle->GetTensorInfo().GetDataType())
{
case DataType::Float16:
- InitialiseArmComputeClTensorData(clTensor, handle->GetConstTensor<armnn::Half>());
+ CopyArmComputeClTensorData(clTensor, handle->GetConstTensor<armnn::Half>());
break;
case DataType::Float32:
- InitialiseArmComputeClTensorData(clTensor, handle->GetConstTensor<float>());
+ CopyArmComputeClTensorData(clTensor, handle->GetConstTensor<float>());
break;
case DataType::QuantisedAsymm8:
- InitialiseArmComputeClTensorData(clTensor, handle->GetConstTensor<uint8_t>());
+ CopyArmComputeClTensorData(clTensor, handle->GetConstTensor<uint8_t>());
break;
case DataType::Signed32:
- InitialiseArmComputeClTensorData(clTensor, handle->GetConstTensor<int32_t>());
+ CopyArmComputeClTensorData(clTensor, handle->GetConstTensor<int32_t>());
break;
default:
BOOST_ASSERT_MSG(false, "Unexpected tensor type.");