From fd627ffaec8fd8801d980b4c91ee7c0607ab6aaf Mon Sep 17 00:00:00 2001 From: Jan Eilers Date: Thu, 25 Feb 2021 17:44:00 +0000 Subject: IVGCVSW-5687 Update Doxygen Docu * Update Doxygen Documentation for 21.02 release Signed-off-by: Jan Eilers Change-Id: I9ed2f9caab038836ea99d7b378d7899fe431a4e5 --- ..._normalization_float_workload_8hpp_source.xhtml | 124 +++++++++++++++++++++ 1 file changed, 124 insertions(+) create mode 100644 21.02/_cl_batch_normalization_float_workload_8hpp_source.xhtml (limited to '21.02/_cl_batch_normalization_float_workload_8hpp_source.xhtml') diff --git a/21.02/_cl_batch_normalization_float_workload_8hpp_source.xhtml b/21.02/_cl_batch_normalization_float_workload_8hpp_source.xhtml new file mode 100644 index 0000000000..e6ae644596 --- /dev/null +++ b/21.02/_cl_batch_normalization_float_workload_8hpp_source.xhtml @@ -0,0 +1,124 @@ + + + + + + + + + + + + + +ArmNN: src/backends/cl/workloads/ClBatchNormalizationFloatWorkload.hpp Source File + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  21.02 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+
ClBatchNormalizationFloatWorkload.hpp
+
+
+Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
8 #include <backendsCommon/Workload.hpp>
9 
10 #include <arm_compute/runtime/CL/CLTensor.h>
11 #include <arm_compute/runtime/CL/functions/CLBatchNormalizationLayer.h>
12 
13 namespace armnn
14 {
15 
16 arm_compute::Status ClBatchNormalizationValidate(const TensorInfo& input,
17  const TensorInfo& output,
18  const TensorInfo& mean,
19  const TensorInfo& var,
20  const TensorInfo& beta,
21  const TensorInfo& gamma,
22  const BatchNormalizationDescriptor& desc,
23  const ActivationDescriptor* activationDescriptor = nullptr);
24 
25 class ClBatchNormalizationFloatWorkload : public FloatWorkload<BatchNormalizationQueueDescriptor>
26 {
27 public:
28  ClBatchNormalizationFloatWorkload(const BatchNormalizationQueueDescriptor& descriptor,
29  const WorkloadInfo& info,
30  const arm_compute::CLCompileContext& clCompileContext);
31 
32  using FloatWorkload<BatchNormalizationQueueDescriptor>::FloatWorkload;
33  void Execute() const override;
34 
35 private:
36  mutable arm_compute::CLBatchNormalizationLayer m_Layer;
37 
38  std::unique_ptr<arm_compute::CLTensor> m_Mean;
39  std::unique_ptr<arm_compute::CLTensor> m_Variance;
40  std::unique_ptr<arm_compute::CLTensor> m_Gamma;
41  std::unique_ptr<arm_compute::CLTensor> m_Beta;
42 
43  void FreeUnusedTensors();
44 };
45 
46 } //namespace armnn
47 
48 
49 
50 
armnn::ClBatchNormalizationFloatWorkload::ClBatchNormalizationFloatWorkload
ClBatchNormalizationFloatWorkload(const BatchNormalizationQueueDescriptor &descriptor, const WorkloadInfo &info, const arm_compute::CLCompileContext &clCompileContext)
Definition: ClBatchNormalizationFloatWorkload.cpp:54
+
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_software_tools.dox:6
+
armnn::Status
Status
enumeration
Definition: Types.hpp:26
+ +
Workload.hpp
+ + + +
armnn::ClBatchNormalizationValidate
arm_compute::Status ClBatchNormalizationValidate(const TensorInfo &input, const TensorInfo &output, const TensorInfo &mean, const TensorInfo &var, const TensorInfo &beta, const TensorInfo &gamma, const BatchNormalizationDescriptor &desc, const ActivationDescriptor *activationDescriptor)
Definition: ClBatchNormalizationFloatWorkload.cpp:19
+
armnn::WorkloadInfo
Contains information about inputs and outputs to a layer.
Definition: WorkloadInfo.hpp:16
+ +
+
+ + + + -- cgit v1.2.1