ArmNN
 24.02
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 {
19  const TensorInfo& indices,
20  const TensorInfo& output);
21 
22 class NeonGatherNdWorkload : public NeonBaseWorkload<GatherNdQueueDescriptor>
23 {
24 public:
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::GatherNdQueueDescriptor
Definition: WorkloadData.hpp:502
NeonBaseWorkload.hpp
armnn::NeonGatherNdWorkload::NeonGatherNdWorkload
NeonGatherNdWorkload(const GatherNdQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition: NeonGatherNdWorkload.cpp:97
armnn::NeonGatherNdWorkloadValidate
arm_compute::Status NeonGatherNdWorkloadValidate(const TensorInfo &paramsInfo, const TensorInfo &indicesInfo, const TensorInfo &outputInfo)
Definition: NeonGatherNdWorkload.cpp:14
armnn::WorkloadInfo
Contains information about TensorInfos of a layer.
Definition: WorkloadInfo.hpp:16
armnn::NeonGatherNdWorkload::Execute
virtual void Execute() const override
Definition: NeonGatherNdWorkload.cpp:190
armnn::BoostLogSeverityMapping::info
@ info
armnn::Status
Status
Definition: Types.hpp:42
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6
armnn::NeonGatherNdWorkload
Definition: NeonGatherNdWorkload.hpp:22
armnn::NeonBaseWorkload
Definition: NeonBaseWorkload.hpp:13