From 3122bd574a3d29774c535ca2136de361da626e88 Mon Sep 17 00:00:00 2001 From: Matteo Martincigh Date: Mon, 3 Jun 2019 16:54:25 +0100 Subject: IVGCVSW-3212 Refactor the Reference BatchNormalization workloads to handle Float32 and QAsymm8 types * Removed the type-specific workload implementations * Added type-independent RefBatchNormalizationWorkload implementation * Reworked BachNormImpl to use decoders/encoders * Improved the validation of the BatchNorm queue descriptor * Fixed unit tests where necessary Change-Id: Icf3fa1332292d38ec2fa0b1cb984cab78426034b Signed-off-by: Matteo Martincigh --- .../RefBatchNormalizationFloat32Workload.cpp | 38 ---------------------- 1 file changed, 38 deletions(-) delete mode 100644 src/backends/reference/workloads/RefBatchNormalizationFloat32Workload.cpp (limited to 'src/backends/reference/workloads/RefBatchNormalizationFloat32Workload.cpp') diff --git a/src/backends/reference/workloads/RefBatchNormalizationFloat32Workload.cpp b/src/backends/reference/workloads/RefBatchNormalizationFloat32Workload.cpp deleted file mode 100644 index 313af9c438..0000000000 --- a/src/backends/reference/workloads/RefBatchNormalizationFloat32Workload.cpp +++ /dev/null @@ -1,38 +0,0 @@ -// -// Copyright © 2017 Arm Ltd. All rights reserved. -// SPDX-License-Identifier: MIT -// - -#include "RefBatchNormalizationFloat32Workload.hpp" - -#include "BatchNormImpl.hpp" -#include "RefWorkloadUtils.hpp" - -#include "Profiling.hpp" - -namespace armnn -{ -RefBatchNormalizationFloat32Workload::RefBatchNormalizationFloat32Workload( - const BatchNormalizationQueueDescriptor& descriptor, const WorkloadInfo& info) - : Float32Workload(descriptor, info), - m_Mean(std::make_unique(*(descriptor.m_Mean))), - m_Variance(std::make_unique(*(descriptor.m_Variance))), - m_Beta(std::make_unique(*(descriptor.m_Beta))), - m_Gamma(std::make_unique(*(descriptor.m_Gamma))) {} - -void RefBatchNormalizationFloat32Workload::Execute() const -{ - ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefBatchNormalizationFloat32Workload_Execute"); - - const float* var = m_Variance->GetConstTensor(); - const float* mean = m_Mean->GetConstTensor(); - const float* gamma = m_Gamma->GetConstTensor(); - const float* beta = m_Beta->GetConstTensor(); - - auto inputData = GetInputTensorDataFloat(0, m_Data); - auto outputData = GetOutputTensorDataFloat(0, m_Data); - - BatchNormImpl(m_Data, var, mean, gamma, beta, outputData, inputData); -} - -} //namespace armnn -- cgit v1.2.1