diff options
Diffstat (limited to 'src/backends/neon/workloads/NeonDepthwiseConvolutionFloatWorkload.cpp')
| -rw-r--r-- | src/backends/neon/workloads/NeonDepthwiseConvolutionFloatWorkload.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/backends/neon/workloads/NeonDepthwiseConvolutionFloatWorkload.cpp b/src/backends/neon/workloads/NeonDepthwiseConvolutionFloatWorkload.cpp index 4b266f3288..9790998ebe 100644 --- a/src/backends/neon/workloads/NeonDepthwiseConvolutionFloatWorkload.cpp +++ b/src/backends/neon/workloads/NeonDepthwiseConvolutionFloatWorkload.cpp @@ -69,11 +69,11 @@ NeonDepthwiseConvolutionFloatWorkload::NeonDepthwiseConvolutionFloatWorkload( BOOST_ASSERT(m_pDepthwiseConvolutionLayer); - InitializeArmComputeTensorDataForFloatTypes(*m_KernelTensor, m_Data.m_Weight); + InitializeArmComputeTensorData(*m_KernelTensor, m_Data.m_Weight); if (m_BiasTensor) { - InitializeArmComputeTensorDataForFloatTypes(*m_BiasTensor, m_Data.m_Bias); + InitializeArmComputeTensorData(*m_BiasTensor, m_Data.m_Bias); } m_pDepthwiseConvolutionLayer->prepare(); |