From c48ac8c8cea1748ebfef15144f070799d4a129c3 Mon Sep 17 00:00:00 2001 From: Matthew Bentham Date: Wed, 12 Dec 2018 16:15:59 +0000 Subject: MLCE-80 Remove strong typing from NeonBatchNormalization Technical debt work towards adding some new Neon workloads Change-Id: I08ab6dd14d0e89d4ebc8a878fb69caa5681012bf --- .../NeonBatchNormalizationFloatWorkload.hpp | 42 ---------------------- 1 file changed, 42 deletions(-) delete mode 100644 src/backends/neon/workloads/NeonBatchNormalizationFloatWorkload.hpp (limited to 'src/backends/neon/workloads/NeonBatchNormalizationFloatWorkload.hpp') diff --git a/src/backends/neon/workloads/NeonBatchNormalizationFloatWorkload.hpp b/src/backends/neon/workloads/NeonBatchNormalizationFloatWorkload.hpp deleted file mode 100644 index a6289bd900..0000000000 --- a/src/backends/neon/workloads/NeonBatchNormalizationFloatWorkload.hpp +++ /dev/null @@ -1,42 +0,0 @@ -// -// Copyright © 2017 Arm Ltd. All rights reserved. -// SPDX-License-Identifier: MIT -// - -#pragma once - -#include - -namespace armnn -{ - -arm_compute::Status NeonBatchNormalizationValidate(const TensorInfo& input, - const TensorInfo& output, - const TensorInfo& mean, - const TensorInfo& var, - const TensorInfo& beta, - const TensorInfo& gamma, - const BatchNormalizationDescriptor& descriptor); - -class NeonBatchNormalizationFloatWorkload : public FloatWorkload -{ -public: - NeonBatchNormalizationFloatWorkload(const BatchNormalizationQueueDescriptor& descriptor, - const WorkloadInfo& info); - virtual void Execute() const override; - -private: - mutable arm_compute::NEBatchNormalizationLayer m_Layer; - - std::unique_ptr m_Mean; - std::unique_ptr m_Variance; - std::unique_ptr m_Gamma; - std::unique_ptr m_Beta; - - void FreeUnusedTensors(); -}; - -} //namespace armnn - - - -- cgit v1.2.1