From ae050524109f1ce827962665436ef7430f2ac479 Mon Sep 17 00:00:00 2001 From: David Monahan Date: Wed, 22 Mar 2023 16:48:58 +0000 Subject: IVGCVSW-7255 Update Doxygen Documentation and publish on GitHub. * Updating Doxygen documentation for 23.02 release. Signed-off-by: David Monahan Change-Id: I545574ff7664b4595d2fe6a91a3c35d2ad55df82 --- 23.02/_neon_gather_nd_workload_8hpp_source.xhtml | 86 ++++++++++++++++++------ 1 file changed, 65 insertions(+), 21 deletions(-) (limited to '23.02/_neon_gather_nd_workload_8hpp_source.xhtml') diff --git a/23.02/_neon_gather_nd_workload_8hpp_source.xhtml b/23.02/_neon_gather_nd_workload_8hpp_source.xhtml index c6d387bbf8..779823d4b5 100644 --- a/23.02/_neon_gather_nd_workload_8hpp_source.xhtml +++ b/23.02/_neon_gather_nd_workload_8hpp_source.xhtml @@ -8,7 +8,7 @@ - + ArmNN: src/backends/neon/workloads/NeonGatherNdWorkload.hpp Source File @@ -19,9 +19,6 @@ - @@ -30,7 +27,8 @@ extensions: ["tex2jax.js"], jax: ["input/TeX","output/HTML-CSS"], }); - + + @@ -51,18 +49,21 @@ - + +/* @license-end */
@@ -76,7 +77,9 @@ $(function() {
@@ -98,26 +101,67 @@ $(document).ready(function(){initNavTree('_neon_gather_nd_workload_8hpp_source.x
NeonGatherNdWorkload.hpp
-Go to the documentation of this file.
1 //
2 // Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
8 #include "NeonBaseWorkload.hpp"
9 
10 #include "arm_compute/runtime/Tensor.h"
11 #include "arm_compute/runtime/NEON/functions/NEGather.h"
12 #include "arm_compute/runtime/NEON/functions/NEPixelWiseMultiplication.h"
13 #include "arm_compute/runtime/NEON/functions/NEReductionOperation.h"
14 #include "arm_compute/runtime/NEON/functions/NEReshapeLayer.h"
15 
16 namespace armnn
17 {
18 arm_compute::Status NeonGatherNdWorkloadValidate(const TensorInfo& input,
19  const TensorInfo& indices,
20  const TensorInfo& output);
21 
22 class NeonGatherNdWorkload : public NeonBaseWorkload<GatherNdQueueDescriptor>
23 {
24 public:
25  NeonGatherNdWorkload(const GatherNdQueueDescriptor& descriptor, const WorkloadInfo& info);
26  virtual void Execute() const override;
27 
28 private:
29  arm_compute::Tensor m_FlattenedCoeff;
30  arm_compute::Tensor m_OutputMul;
31  arm_compute::Tensor m_FlattenedIndices;
32  arm_compute::Tensor m_OutputGather;
33 
34  mutable arm_compute::NEPixelWiseMultiplication m_MulLayer;
35  mutable arm_compute::NEReductionOperation m_ReduceSumLayer;
36  mutable arm_compute::NEGather m_GatherLayer;
37  mutable arm_compute::NEReshapeLayer m_ReshapeLayer;
38 
39 };
40 
41 } //namespace armnn
armnn::NeonGatherNdWorkloadValidate
arm_compute::Status NeonGatherNdWorkloadValidate(const TensorInfo &paramsInfo, const TensorInfo &indicesInfo, const TensorInfo &outputInfo)
Definition: NeonGatherNdWorkload.cpp:14
- -
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6
- -
armnn::NeonGatherNdWorkload::Execute
virtual void Execute() const override
Definition: NeonGatherNdWorkload.cpp:190
-
armnn::Status
Status
enumeration
Definition: Types.hpp:42
-
armnn::NeonGatherNdWorkload::NeonGatherNdWorkload
NeonGatherNdWorkload(const GatherNdQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition: NeonGatherNdWorkload.cpp:97
- - -
NeonBaseWorkload.hpp
-
armnn::WorkloadInfo
Contains information about TensorInfos of a layer.
Definition: WorkloadInfo.hpp:16
+Go to the documentation of this file.
1 //
+
2 // Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
+
3 // SPDX-License-Identifier: MIT
+
4 //
+
5 
+
6 #pragma once
+
7 
+
8 #include "NeonBaseWorkload.hpp"
+
9 
+
10 #include "arm_compute/runtime/Tensor.h"
+
11 #include "arm_compute/runtime/NEON/functions/NEGather.h"
+
12 #include "arm_compute/runtime/NEON/functions/NEPixelWiseMultiplication.h"
+
13 #include "arm_compute/runtime/NEON/functions/NEReductionOperation.h"
+
14 #include "arm_compute/runtime/NEON/functions/NEReshapeLayer.h"
+
15 
+
16 namespace armnn
+
17 {
+
18 arm_compute::Status NeonGatherNdWorkloadValidate(const TensorInfo& input,
+
19  const TensorInfo& indices,
+
20  const TensorInfo& output);
+
21 
+
22 class NeonGatherNdWorkload : public NeonBaseWorkload<GatherNdQueueDescriptor>
+
23 {
+
24 public:
+
25  NeonGatherNdWorkload(const GatherNdQueueDescriptor& descriptor, const WorkloadInfo& info);
+
26  virtual void Execute() const override;
+
27 
+
28 private:
+
29  arm_compute::Tensor m_FlattenedCoeff;
+
30  arm_compute::Tensor m_OutputMul;
+
31  arm_compute::Tensor m_FlattenedIndices;
+
32  arm_compute::Tensor m_OutputGather;
+
33 
+
34  mutable arm_compute::NEPixelWiseMultiplication m_MulLayer;
+
35  mutable arm_compute::NEReductionOperation m_ReduceSumLayer;
+
36  mutable arm_compute::NEGather m_GatherLayer;
+
37  mutable arm_compute::NEReshapeLayer m_ReshapeLayer;
+
38 
+
39 };
+
40 
+
41 } //namespace armnn
+
armnn::NeonGatherNdWorkloadValidate
arm_compute::Status NeonGatherNdWorkloadValidate(const TensorInfo &paramsInfo, const TensorInfo &indicesInfo, const TensorInfo &outputInfo)
Definition: NeonGatherNdWorkload.cpp:14
+
armnn::NeonGatherNdWorkload::Execute
virtual void Execute() const override
Definition: NeonGatherNdWorkload.cpp:190
+ +
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6
+
armnn::Status
Status
Definition: Types.hpp:42
+
armnn::WorkloadInfo
Contains information about TensorInfos of a layer.
Definition: WorkloadInfo.hpp:16
+ + +
armnn::NeonGatherNdWorkload::NeonGatherNdWorkload
NeonGatherNdWorkload(const GatherNdQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition: NeonGatherNdWorkload.cpp:97
+ + -- cgit v1.2.1