From 177d8d26925a58a579943e010d28d1ceaa033d64 Mon Sep 17 00:00:00 2001 From: Nattapat Chaimanowong Date: Tue, 16 Oct 2018 13:21:27 +0100 Subject: IVGCVSW-1951 Update NeonWorkloadUtils Change-Id: I147dbf6811f84ec4588264d636a36efc8ec56f72 --- .../neon/workloads/NeonBatchNormalizationFloatWorkload.cpp | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) (limited to 'src/backends/neon/workloads/NeonBatchNormalizationFloatWorkload.cpp') diff --git a/src/backends/neon/workloads/NeonBatchNormalizationFloatWorkload.cpp b/src/backends/neon/workloads/NeonBatchNormalizationFloatWorkload.cpp index 2383e78df3..f7056a515b 100644 --- a/src/backends/neon/workloads/NeonBatchNormalizationFloatWorkload.cpp +++ b/src/backends/neon/workloads/NeonBatchNormalizationFloatWorkload.cpp @@ -66,10 +66,10 @@ NeonBatchNormalizationFloatWorkload::NeonBatchNormalizationFloatWorkload( m_Gamma.get(), m_Data.m_Parameters.m_Eps); - InitializeArmComputeTensorDataForFloatTypes(*m_Mean, m_Data.m_Mean); - InitializeArmComputeTensorDataForFloatTypes(*m_Variance, m_Data.m_Variance); - InitializeArmComputeTensorDataForFloatTypes(*m_Gamma, m_Data.m_Gamma); - InitializeArmComputeTensorDataForFloatTypes(*m_Beta, m_Data.m_Beta); + InitializeArmComputeTensorData(*m_Mean, m_Data.m_Mean); + InitializeArmComputeTensorData(*m_Variance, m_Data.m_Variance); + InitializeArmComputeTensorData(*m_Gamma, m_Data.m_Gamma); + InitializeArmComputeTensorData(*m_Beta, m_Data.m_Beta); // Force Compute Library to perform the necessary copying and reshaping, after which // delete all the input tensors that will no longer be needed @@ -92,5 +92,3 @@ void NeonBatchNormalizationFloatWorkload::FreeUnusedTensors() } } //namespace armnn - - -- cgit v1.2.1