diff options
| author | arovir01 <Aron.Virginas-Tar@arm.com> | 2018-08-31 15:26:35 +0100 |
|---|---|---|
| committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-09-17 17:21:25 +0100 |
| commit | 9e53a35b66b1ec7ceee7c712380a13596175b83b (patch) | |
| tree | d40bf9f27c799184324b6ab91cbb1a546fc4012e /src/armnn/backends/NeonWorkloads/NeonBatchNormalizationFloat32Workload.hpp | |
| parent | 5540d2f379b15503269d1b9b5fbe8fbafd160d2e (diff) | |
| download | armnn-9e53a35b66b1ec7ceee7c712380a13596175b83b.tar.gz | |
IVGCVSW-1784: Rename float32 workloads for ACL
Change-Id: I98bdfe9cb12c663d1d5cfa456e2cc967d70ab22b
Diffstat (limited to 'src/armnn/backends/NeonWorkloads/NeonBatchNormalizationFloat32Workload.hpp')
| -rw-r--r-- | src/armnn/backends/NeonWorkloads/NeonBatchNormalizationFloat32Workload.hpp | 42 |
1 files changed, 0 insertions, 42 deletions
diff --git a/src/armnn/backends/NeonWorkloads/NeonBatchNormalizationFloat32Workload.hpp b/src/armnn/backends/NeonWorkloads/NeonBatchNormalizationFloat32Workload.hpp deleted file mode 100644 index 5eb5601f26..0000000000 --- a/src/armnn/backends/NeonWorkloads/NeonBatchNormalizationFloat32Workload.hpp +++ /dev/null @@ -1,42 +0,0 @@ -// -// Copyright © 2017 Arm Ltd. All rights reserved. -// See LICENSE file in the project root for full license information. -// - -#pragma once - -#include <backends/NeonWorkloadUtils.hpp> - -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 NeonBatchNormalizationFloat32Workload : public FloatWorkload<BatchNormalizationQueueDescriptor> -{ -public: - NeonBatchNormalizationFloat32Workload(const BatchNormalizationQueueDescriptor& descriptor, - const WorkloadInfo& info); - virtual void Execute() const override; - -private: - mutable arm_compute::NEBatchNormalizationLayer m_Layer; - - std::unique_ptr<arm_compute::Tensor> m_Mean; - std::unique_ptr<arm_compute::Tensor> m_Variance; - std::unique_ptr<arm_compute::Tensor> m_Gamma; - std::unique_ptr<arm_compute::Tensor> m_Beta; - - void FreeUnusedTensors(); -}; - -} //namespace armnn - - - |