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/NeonDepthwiseConvolutionUint8Workload.cpp | |
| parent | 97a06fd57e7864a882ef5e37a1bf7286f5be5185 (diff) | |
| download | armnn-177d8d26925a58a579943e010d28d1ceaa033d64.tar.gz | |
IVGCVSW-1951 Update NeonWorkloadUtils
Change-Id: I147dbf6811f84ec4588264d636a36efc8ec56f72
Diffstat (limited to 'src/backends/neon/workloads/NeonDepthwiseConvolutionUint8Workload.cpp')
| -rw-r--r-- | src/backends/neon/workloads/NeonDepthwiseConvolutionUint8Workload.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/backends/neon/workloads/NeonDepthwiseConvolutionUint8Workload.cpp b/src/backends/neon/workloads/NeonDepthwiseConvolutionUint8Workload.cpp index 6c6c2dfb6c..25d00f92ca 100644 --- a/src/backends/neon/workloads/NeonDepthwiseConvolutionUint8Workload.cpp +++ b/src/backends/neon/workloads/NeonDepthwiseConvolutionUint8Workload.cpp @@ -69,11 +69,11 @@ NeonDepthwiseConvolutionUint8Workload::NeonDepthwiseConvolutionUint8Workload( BOOST_ASSERT(m_pDepthwiseConvolutionLayer); - InitialiseArmComputeTensorData(*m_KernelTensor, m_Data.m_Weight->GetConstTensor<uint8_t>()); + InitializeArmComputeTensorData(*m_KernelTensor, m_Data.m_Weight); if (m_BiasTensor) { - InitialiseArmComputeTensorData(*m_BiasTensor, m_Data.m_Bias->GetConstTensor<int32_t>()); + InitializeArmComputeTensorData(*m_BiasTensor, m_Data.m_Bias); } m_pDepthwiseConvolutionLayer->prepare(); |