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" 19 const TensorInfo& indices,
20 const TensorInfo& output);
26 virtual void Execute()
const override;
29 arm_compute::Tensor m_FlattenedCoeff;
30 arm_compute::Tensor m_OutputMul;
31 arm_compute::Tensor m_FlattenedIndices;
32 arm_compute::Tensor m_OutputGather;
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;
arm_compute::Status NeonGatherNdWorkloadValidate(const TensorInfo ¶msInfo, const TensorInfo &indicesInfo, const TensorInfo &outputInfo)
Copyright (c) 2021 ARM Limited and Contributors.
virtual void Execute() const override
NeonGatherNdWorkload(const GatherNdQueueDescriptor &descriptor, const WorkloadInfo &info)
Contains information about TensorInfos of a layer.