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/ClFullyConnectedWorkload.cpp | |
parent | 10b4dfd8e9ccd7a03df7bb053ee1c644cb37f8ab (diff) | |
download | armnn-785df505a621a5b98084246056d80090073b950b.tar.gz |
IVGCVSW-949 Simplify use of IntialiseArmComputeClTensorData
Change-Id: I556881e34f26e8152feaaba06d99828394872f58
Diffstat (limited to 'src/backends/ClWorkloads/ClFullyConnectedWorkload.cpp')
-rw-r--r-- | src/backends/ClWorkloads/ClFullyConnectedWorkload.cpp | 19 |
1 files changed, 2 insertions, 17 deletions
diff --git a/src/backends/ClWorkloads/ClFullyConnectedWorkload.cpp b/src/backends/ClWorkloads/ClFullyConnectedWorkload.cpp index 8d2fd0e909..4686d1c8ee 100644 --- a/src/backends/ClWorkloads/ClFullyConnectedWorkload.cpp +++ b/src/backends/ClWorkloads/ClFullyConnectedWorkload.cpp @@ -68,26 +68,11 @@ ClFullyConnectedWorkload::ClFullyConnectedWorkload(const FullyConnectedQueueDesc fc_info.transpose_weights = m_Data.m_Parameters.m_TransposeWeightMatrix; m_FullyConnectedLayer.configure(&input, m_WeightsTensor.get(), m_BiasesTensor.get(), &output, fc_info); - // Allocate - if (m_Data.m_Weight->GetTensorInfo().GetDataType() == DataType::QuantisedAsymm8) - { - InitialiseArmComputeClTensorData(*m_WeightsTensor, m_Data.m_Weight->GetConstTensor<uint8_t>()); - } - else - { - InitializeArmComputeClTensorDataForFloatTypes(*m_WeightsTensor, m_Data.m_Weight); - } + InitializeArmComputeClTensorData(*m_WeightsTensor, m_Data.m_Weight); if (m_BiasesTensor) { - if (m_Data.m_Bias->GetTensorInfo().GetDataType() == DataType::Signed32) - { - InitialiseArmComputeClTensorData(*m_BiasesTensor, m_Data.m_Bias->GetConstTensor<int32_t>()); - } - else - { - InitializeArmComputeClTensorDataForFloatTypes(*m_BiasesTensor, m_Data.m_Bias); - } + InitializeArmComputeClTensorData(*m_BiasesTensor, m_Data.m_Bias); } // Force Compute Library to perform the necessary copying and reshaping, after which |