diff options
author | Matthew Bentham <matthew.bentham@arm.com> | 2018-12-12 16:15:59 +0000 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-12-31 15:56:48 +0000 |
commit | c48ac8c8cea1748ebfef15144f070799d4a129c3 (patch) | |
tree | 62eead8b1d684fa7edbd3e2a1a70e4ed871a1f30 /src/backends/neon/workloads | |
parent | fbdad03c927aa5d30deec6fa1a61eef10f8c265f (diff) | |
download | armnn-c48ac8c8cea1748ebfef15144f070799d4a129c3.tar.gz |
MLCE-80 Remove strong typing from NeonBatchNormalization
Technical debt work towards adding some new Neon workloads
Change-Id: I08ab6dd14d0e89d4ebc8a878fb69caa5681012bf
Diffstat (limited to 'src/backends/neon/workloads')
-rw-r--r-- | src/backends/neon/workloads/CMakeLists.txt | 4 | ||||
-rw-r--r-- | src/backends/neon/workloads/NeonBatchNormalizationWorkload.cpp (renamed from src/backends/neon/workloads/NeonBatchNormalizationFloatWorkload.cpp) | 14 | ||||
-rw-r--r-- | src/backends/neon/workloads/NeonBatchNormalizationWorkload.hpp (renamed from src/backends/neon/workloads/NeonBatchNormalizationFloatWorkload.hpp) | 8 | ||||
-rw-r--r-- | src/backends/neon/workloads/NeonWorkloads.hpp | 2 |
4 files changed, 13 insertions, 15 deletions
diff --git a/src/backends/neon/workloads/CMakeLists.txt b/src/backends/neon/workloads/CMakeLists.txt index 9961397445..7b0251ce04 100644 --- a/src/backends/neon/workloads/CMakeLists.txt +++ b/src/backends/neon/workloads/CMakeLists.txt @@ -8,8 +8,8 @@ list(APPEND armnnNeonBackendWorkloads_sources NeonActivationWorkload.hpp NeonAdditionWorkload.cpp NeonAdditionWorkload.hpp - NeonBatchNormalizationFloatWorkload.cpp - NeonBatchNormalizationFloatWorkload.hpp + NeonBatchNormalizationWorkload.cpp + NeonBatchNormalizationWorkload.hpp NeonConstantWorkload.cpp NeonConstantWorkload.hpp NeonConvertFp16ToFp32Workload.cpp diff --git a/src/backends/neon/workloads/NeonBatchNormalizationFloatWorkload.cpp b/src/backends/neon/workloads/NeonBatchNormalizationWorkload.cpp index a8181f66d9..44d5035431 100644 --- a/src/backends/neon/workloads/NeonBatchNormalizationFloatWorkload.cpp +++ b/src/backends/neon/workloads/NeonBatchNormalizationWorkload.cpp @@ -3,7 +3,7 @@ // SPDX-License-Identifier: MIT // -#include "NeonBatchNormalizationFloatWorkload.hpp" +#include "NeonBatchNormalizationWorkload.hpp" #include <backendsCommon/CpuTensorHandle.hpp> #include <aclCommon/ArmComputeTensorUtils.hpp> #include <armnn/ArmNN.hpp> @@ -43,11 +43,11 @@ arm_compute::Status NeonBatchNormalizationValidate(const TensorInfo& input, descriptor.m_Eps); } -NeonBatchNormalizationFloatWorkload::NeonBatchNormalizationFloatWorkload( +NeonBatchNormalizationWorkload::NeonBatchNormalizationWorkload( const BatchNormalizationQueueDescriptor& descriptor, const WorkloadInfo& info) - : FloatWorkload<BatchNormalizationQueueDescriptor>(descriptor, info) + : BaseWorkload<BatchNormalizationQueueDescriptor>(descriptor, info) { - m_Data.ValidateInputsOutputs("NeonBatchNormalizationFloatWorkload", 1, 1); + m_Data.ValidateInputsOutputs("NeonBatchNormalizationWorkload", 1, 1); arm_compute::ITensor& input = boost::polymorphic_downcast<INeonTensorHandle*>(m_Data.m_Inputs[0])->GetTensor(); arm_compute::ITensor& output = boost::polymorphic_downcast<INeonTensorHandle*>(m_Data.m_Outputs[0])->GetTensor(); @@ -87,13 +87,13 @@ NeonBatchNormalizationFloatWorkload::NeonBatchNormalizationFloatWorkload( FreeUnusedTensors(); } -void NeonBatchNormalizationFloatWorkload::Execute() const +void NeonBatchNormalizationWorkload::Execute() const { - ARMNN_SCOPED_PROFILING_EVENT_NEON("NeonBatchNormalizationFloatWorkload_Execute"); + ARMNN_SCOPED_PROFILING_EVENT_NEON("NeonBatchNormalizationWorkload_Execute"); m_Layer.run(); } -void NeonBatchNormalizationFloatWorkload::FreeUnusedTensors() +void NeonBatchNormalizationWorkload::FreeUnusedTensors() { FreeTensorIfUnused(m_Mean); FreeTensorIfUnused(m_Variance); diff --git a/src/backends/neon/workloads/NeonBatchNormalizationFloatWorkload.hpp b/src/backends/neon/workloads/NeonBatchNormalizationWorkload.hpp index a6289bd900..52e4db7c90 100644 --- a/src/backends/neon/workloads/NeonBatchNormalizationFloatWorkload.hpp +++ b/src/backends/neon/workloads/NeonBatchNormalizationWorkload.hpp @@ -18,11 +18,11 @@ arm_compute::Status NeonBatchNormalizationValidate(const TensorInfo& input, const TensorInfo& gamma, const BatchNormalizationDescriptor& descriptor); -class NeonBatchNormalizationFloatWorkload : public FloatWorkload<BatchNormalizationQueueDescriptor> +class NeonBatchNormalizationWorkload : public BaseWorkload<BatchNormalizationQueueDescriptor> { public: - NeonBatchNormalizationFloatWorkload(const BatchNormalizationQueueDescriptor& descriptor, - const WorkloadInfo& info); + NeonBatchNormalizationWorkload(const BatchNormalizationQueueDescriptor& descriptor, + const WorkloadInfo& info); virtual void Execute() const override; private: @@ -38,5 +38,3 @@ private: } //namespace armnn - - diff --git a/src/backends/neon/workloads/NeonWorkloads.hpp b/src/backends/neon/workloads/NeonWorkloads.hpp index 82142f2aa1..1f08d039ae 100644 --- a/src/backends/neon/workloads/NeonWorkloads.hpp +++ b/src/backends/neon/workloads/NeonWorkloads.hpp @@ -6,7 +6,7 @@ #pragma once #include "NeonActivationWorkload.hpp" #include "NeonAdditionWorkload.hpp" -#include "NeonBatchNormalizationFloatWorkload.hpp" +#include "NeonBatchNormalizationWorkload.hpp" #include "NeonConstantWorkload.hpp" #include "NeonConvertFp16ToFp32Workload.hpp" #include "NeonConvertFp32ToFp16Workload.hpp" |