diff options
| author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2018-10-16 13:21:27 +0100 |
|---|---|---|
| committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-22 16:57:54 +0100 |
| commit | 177d8d26925a58a579943e010d28d1ceaa033d64 (patch) | |
| tree | 73bc6bb530621f33a6b7cd27ae303a3f92fb556c /src/backends/neon/workloads/NeonConvolution2dBaseWorkload.cpp | |
| parent | 97a06fd57e7864a882ef5e37a1bf7286f5be5185 (diff) | |
| download | armnn-177d8d26925a58a579943e010d28d1ceaa033d64.tar.gz | |
IVGCVSW-1951 Update NeonWorkloadUtils
Change-Id: I147dbf6811f84ec4588264d636a36efc8ec56f72
Diffstat (limited to 'src/backends/neon/workloads/NeonConvolution2dBaseWorkload.cpp')
| -rw-r--r-- | src/backends/neon/workloads/NeonConvolution2dBaseWorkload.cpp | 25 |
1 files changed, 1 insertions, 24 deletions
diff --git a/src/backends/neon/workloads/NeonConvolution2dBaseWorkload.cpp b/src/backends/neon/workloads/NeonConvolution2dBaseWorkload.cpp index 8da3e47249..b11d10fd2f 100644 --- a/src/backends/neon/workloads/NeonConvolution2dBaseWorkload.cpp +++ b/src/backends/neon/workloads/NeonConvolution2dBaseWorkload.cpp @@ -109,30 +109,8 @@ NeonConvolution2dBaseWorkload<dataTypes...>::NeonConvolution2dBaseWorkload( } BOOST_ASSERT(m_ConvolutionLayer); - armnn::DataType dataType = m_Data.m_Weight->GetTensorInfo().GetDataType(); + InitializeArmComputeTensorData(*m_KernelTensor, m_Data.m_Weight); - switch (dataType) - { - case DataType::Float16: - { - InitialiseArmComputeTensorData(*m_KernelTensor, m_Data.m_Weight->template GetConstTensor<Half>()); - break; - } - case DataType::Float32: - { - InitialiseArmComputeTensorData(*m_KernelTensor, m_Data.m_Weight->template GetConstTensor<float>()); - break; - } - case DataType::QuantisedAsymm8: - { - InitialiseArmComputeTensorData(*m_KernelTensor, m_Data.m_Weight->template GetConstTensor<uint8_t>()); - break; - } - default: - { - BOOST_ASSERT_MSG(false, "Unknown DataType."); - } - } } template<armnn::DataType... dataTypes> @@ -147,4 +125,3 @@ template class NeonConvolution2dBaseWorkload<armnn::DataType::Float16, armnn::Da template class NeonConvolution2dBaseWorkload<armnn::DataType::QuantisedAsymm8>; } //namespace armnn - |