ArmNN
 24.02
ClNormalizationFloatWorkload.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
9 
10 #include <arm_compute/runtime/CL/functions/CLNormalizationLayer.h>
11 
12 namespace armnn
13 {
14 
16  const TensorInfo& output,
17  const NormalizationDescriptor& descriptor);
18 
19 class ClNormalizationFloatWorkload : public FloatWorkload<NormalizationQueueDescriptor>
20 {
21 public:
23  const WorkloadInfo& info,
24  const arm_compute::CLCompileContext& clCompileContext);
25  void Execute() const override;
26  // Replace input tensor handle with the given TensorHandle
27  void ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override;
28 
29  // Replace output tensor handle with the given TensorHandle
30  void ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override;
31 private:
32  mutable arm_compute::CLNormalizationLayer m_NormalizationLayer;
33  virtual void Reconfigure();
34 };
35 
36 } //namespace armnn
armnn::NormalizationQueueDescriptor
Definition: WorkloadData.hpp:252
armnn::ClNormalizationFloatWorkload::ReplaceInputTensorHandle
void ReplaceInputTensorHandle(ITensorHandle *tensorHandle, unsigned int slot) override
Definition: ClNormalizationFloatWorkload.cpp:65
armnn::ITensorHandle
Definition: ITensorHandle.hpp:16
armnn::TypedWorkload
Definition: Workload.hpp:101
armnn::WorkloadInfo
Contains information about TensorInfos of a layer.
Definition: WorkloadInfo.hpp:16
Workload.hpp
armnn::ClNormalizationFloatWorkload::Execute
void Execute() const override
Definition: ClNormalizationFloatWorkload.cpp:59
armnn::ClNormalizationFloatWorkload::ReplaceOutputTensorHandle
void ReplaceOutputTensorHandle(ITensorHandle *tensorHandle, unsigned int slot) override
Definition: ClNormalizationFloatWorkload.cpp:82
armnn::BoostLogSeverityMapping::info
@ info
armnn::ClNormalizationWorkloadValidate
arm_compute::Status ClNormalizationWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const NormalizationDescriptor &descriptor)
Definition: ClNormalizationFloatWorkload.cpp:19
armnn::Status
Status
Definition: Types.hpp:42
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6
armnn::ClNormalizationFloatWorkload
Definition: ClNormalizationFloatWorkload.hpp:19
armnn::ClNormalizationFloatWorkload::ClNormalizationFloatWorkload
ClNormalizationFloatWorkload(const NormalizationQueueDescriptor &descriptor, const WorkloadInfo &info, const arm_compute::CLCompileContext &clCompileContext)
Definition: ClNormalizationFloatWorkload.cpp:31