diff options
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(); |