src/backends/neon/workloads/NeonNormalizationFloatWorkload.hpp

   1 //
   2 // Copyright © 2017 Arm Ltd. All rights reserved.
   3 // SPDX-License-Identifier: MIT
   4 //
   5
   6 #pragma once
   7
   8 #include <neon/workloads/NeonWorkloadUtils.hpp>
   9 #include <arm_compute/runtime/MemoryManagerOnDemand.h>
  10
  11 namespace armnn
  12 {
  13
  14 arm_compute::Status NeonNormalizationWorkloadValidate(const TensorInfo& input,
  15                                                       const TensorInfo& output,
  16                                                       const NormalizationDescriptor& descriptor);
  17
  18 class NeonNormalizationFloatWorkload : public FloatWorkload<NormalizationQueueDescriptor>
  19 {
  20 public:
  21     NeonNormalizationFloatWorkload(const NormalizationQueueDescriptor& descriptor, const WorkloadInfo& info,
  22                                    std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager);
  23     virtual void Execute() const override;
  24
  25 private:
  26     mutable arm_compute::NENormalizationLayer m_NormalizationLayer;
  27 };
  28
  29 } //namespace armnn
  30
  31
  32
  33