From 9aed8fb43441228343b925b42464a55042c47ca0 Mon Sep 17 00:00:00 2001 From: Nikhil Raj Date: Wed, 17 Nov 2021 13:16:45 +0000 Subject: IVGCVSW-6040 Update 21.11 Doxygen Documents Signed-off-by: Nikhil Raj Change-Id: Ia36ec98c4bebc27a69103911ea3409cd7db587a5 --- .../_neon_convolution2d_workload_8hpp_source.xhtml | 125 +++++++++++++++++++++ 1 file changed, 125 insertions(+) create mode 100644 21.11/_neon_convolution2d_workload_8hpp_source.xhtml (limited to '21.11/_neon_convolution2d_workload_8hpp_source.xhtml') diff --git a/21.11/_neon_convolution2d_workload_8hpp_source.xhtml b/21.11/_neon_convolution2d_workload_8hpp_source.xhtml new file mode 100644 index 0000000000..21e1c85a49 --- /dev/null +++ b/21.11/_neon_convolution2d_workload_8hpp_source.xhtml @@ -0,0 +1,125 @@ + + + + + + + + + + + + + +ArmNN: src/backends/neon/workloads/NeonConvolution2dWorkload.hpp Source File + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  21.11 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+
NeonConvolution2dWorkload.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/IFunction.h>
11 #include <arm_compute/runtime/Tensor.h>
12 #include <arm_compute/runtime/MemoryManagerOnDemand.h>
13 
14 #include <memory>
15 
16 namespace armnn
17 {
18 
19 arm_compute::Status NeonConvolution2dWorkloadValidate(const TensorInfo& input,
20  const TensorInfo& output,
21  const Convolution2dDescriptor& descriptor,
22  const TensorInfo& weights,
23  const Optional<TensorInfo>& biases,
24  bool isFastMathEnabled = false,
25  const ActivationDescriptor* activationDescriptor = nullptr);
26 
27 class NeonConvolution2dWorkload : public BaseWorkload<Convolution2dQueueDescriptor>
28 {
29 public:
30  using BaseWorkload<Convolution2dQueueDescriptor>::m_Data;
31 
32  NeonConvolution2dWorkload(const Convolution2dQueueDescriptor& descriptor,
33  const WorkloadInfo& info,
34  std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager,
35  const bool isFastMathENabled = false);
36 
37  void Execute() const override;
38 
39  arm_compute::ConvolutionMethod GetConvolutionMethod() const;
40 
41 private:
42  std::unique_ptr<arm_compute::IFunction> m_ConvolutionLayer;
43 
44  std::unique_ptr<arm_compute::Tensor> m_KernelTensor;
45  std::unique_ptr<arm_compute::Tensor> m_BiasTensor;
46 
47  arm_compute::ConvolutionMethod m_ConvolutionMethod;
48 
49  void FreeUnusedTensors();
50 
51 };
52 
53 } //namespace armnn
+
armnn::NeonConvolution2dWorkload::NeonConvolution2dWorkload
NeonConvolution2dWorkload(const Convolution2dQueueDescriptor &descriptor, const WorkloadInfo &info, std::shared_ptr< arm_compute::MemoryManagerOnDemand > &memoryManager, const bool isFastMathENabled=false)
Definition: NeonConvolution2dWorkload.cpp:66
+
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6
+ + +
armnn::Status
Status
enumeration
Definition: Types.hpp:29
+
armnn::NeonConvolution2dWorkloadValidate
arm_compute::Status NeonConvolution2dWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const Convolution2dDescriptor &descriptor, const TensorInfo &weights, const Optional< TensorInfo > &biases, bool isFastMathEnabled, const ActivationDescriptor *activationDescriptor)
Definition: NeonConvolution2dWorkload.cpp:24
+
armnn::NeonConvolution2dWorkload::GetConvolutionMethod
arm_compute::ConvolutionMethod GetConvolutionMethod() const
Definition: NeonConvolution2dWorkload.cpp:160
+ +
Workload.hpp
+ +
armnn::WorkloadInfo
Contains information about TensorInfos of a layer.
Definition: WorkloadInfo.hpp:16
+
+
+ + + + -- cgit v1.2.1