diff options
author | Matthew Bentham <Matthew.Bentham@arm.com> | 2018-09-21 10:29:58 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-10 16:16:56 +0100 |
commit | 785df505a621a5b98084246056d80090073b950b (patch) | |
tree | 506f3de8a1c692ec90aaf81c88f593725579f8b2 /src/backends/ClWorkloads/ClWorkloadUtils.hpp | |
parent | 10b4dfd8e9ccd7a03df7bb053ee1c644cb37f8ab (diff) | |
download | armnn-785df505a621a5b98084246056d80090073b950b.tar.gz |
IVGCVSW-949 Simplify use of IntialiseArmComputeClTensorData
Change-Id: I556881e34f26e8152feaaba06d99828394872f58
Diffstat (limited to 'src/backends/ClWorkloads/ClWorkloadUtils.hpp')
-rw-r--r-- | src/backends/ClWorkloads/ClWorkloadUtils.hpp | 12 |
1 files changed, 9 insertions, 3 deletions
diff --git a/src/backends/ClWorkloads/ClWorkloadUtils.hpp b/src/backends/ClWorkloads/ClWorkloadUtils.hpp index 6f1b155745..a10237cf40 100644 --- a/src/backends/ClWorkloads/ClWorkloadUtils.hpp +++ b/src/backends/ClWorkloads/ClWorkloadUtils.hpp @@ -42,8 +42,8 @@ void InitialiseArmComputeClTensorData(arm_compute::CLTensor& clTensor, const T* CopyArmComputeClTensorData<T>(data, clTensor); } -inline void InitializeArmComputeClTensorDataForFloatTypes(arm_compute::CLTensor& clTensor, - const ConstCpuTensorHandle *handle) +inline void InitializeArmComputeClTensorData(arm_compute::CLTensor& clTensor, + const ConstCpuTensorHandle* handle) { BOOST_ASSERT(handle); switch(handle->GetTensorInfo().GetDataType()) @@ -54,8 +54,14 @@ inline void InitializeArmComputeClTensorDataForFloatTypes(arm_compute::CLTensor& case DataType::Float32: InitialiseArmComputeClTensorData(clTensor, handle->GetConstTensor<float>()); break; + case DataType::QuantisedAsymm8: + InitialiseArmComputeClTensorData(clTensor, handle->GetConstTensor<uint8_t>()); + break; + case DataType::Signed32: + InitialiseArmComputeClTensorData(clTensor, handle->GetConstTensor<int32_t>()); + break; default: - BOOST_ASSERT_MSG(false, "Unexpected floating point type."); + BOOST_ASSERT_MSG(false, "Unexpected tensor type."); } }; |