From ca6616cb6335de2135864f9795f02023a12cda3b Mon Sep 17 00:00:00 2001 From: Matthew Bentham Date: Fri, 21 Sep 2018 15:16:53 +0100 Subject: IVGCVSW-949 Further simplify InitializeArmComputeClTensorData Remove the overload that takes a raw pointer and always pass a ConstTensor reference Change-Id: I8fbff8dd5c260e7cdf4a3c51eb54cc5f3f6e874c --- src/backends/ClWorkloads/ClBaseConstantWorkload.cpp | 6 +++--- src/backends/ClWorkloads/ClWorkloadUtils.hpp | 19 +++++++------------ 2 files changed, 10 insertions(+), 15 deletions(-) (limited to 'src') diff --git a/src/backends/ClWorkloads/ClBaseConstantWorkload.cpp b/src/backends/ClWorkloads/ClBaseConstantWorkload.cpp index 021d17512f..97ddfd6352 100644 --- a/src/backends/ClWorkloads/ClBaseConstantWorkload.cpp +++ b/src/backends/ClWorkloads/ClBaseConstantWorkload.cpp @@ -36,17 +36,17 @@ void ClBaseConstantWorkload::Execute() const { case arm_compute::DataType::F16: { - CopyArmComputeClTensorData(data.m_LayerOutput->GetConstTensor(), output); + CopyArmComputeClTensorData(output, data.m_LayerOutput->GetConstTensor()); break; } case arm_compute::DataType::F32: { - CopyArmComputeClTensorData(data.m_LayerOutput->GetConstTensor(), output); + CopyArmComputeClTensorData(output, data.m_LayerOutput->GetConstTensor()); break; } case arm_compute::DataType::QASYMM8: { - CopyArmComputeClTensorData(data.m_LayerOutput->GetConstTensor(), output); + CopyArmComputeClTensorData(output, data.m_LayerOutput->GetConstTensor()); break; } default: 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 -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 -void InitialiseArmComputeClTensorData(arm_compute::CLTensor& clTensor, const T* data) -{ - armcomputetensorutils::InitialiseArmComputeTensorEmpty(clTensor); - CopyArmComputeClTensorData(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()); + CopyArmComputeClTensorData(clTensor, handle->GetConstTensor()); break; case DataType::Float32: - InitialiseArmComputeClTensorData(clTensor, handle->GetConstTensor()); + CopyArmComputeClTensorData(clTensor, handle->GetConstTensor()); break; case DataType::QuantisedAsymm8: - InitialiseArmComputeClTensorData(clTensor, handle->GetConstTensor()); + CopyArmComputeClTensorData(clTensor, handle->GetConstTensor()); break; case DataType::Signed32: - InitialiseArmComputeClTensorData(clTensor, handle->GetConstTensor()); + CopyArmComputeClTensorData(clTensor, handle->GetConstTensor()); break; default: BOOST_ASSERT_MSG(false, "Unexpected tensor type."); -- cgit v1.2.1