10 #include "arm_compute/runtime/Tensor.h" 11 #include "arm_compute/runtime/CL/functions/CLGather.h" 12 #include "arm_compute/runtime/CL/functions/CLPixelWiseMultiplication.h" 13 #include "arm_compute/runtime/CL/functions/CLReductionOperation.h" 14 #include "arm_compute/runtime/CL/functions/CLReshapeLayer.h" 19 const TensorInfo& indices,
20 const TensorInfo& output);
27 const arm_compute::CLCompileContext& clCompileContext);
28 virtual void Execute()
const override;
31 arm_compute::CLTensor m_FlattenedCoeff;
32 arm_compute::CLTensor m_OutputMul;
33 arm_compute::CLTensor m_FlattenedIndices;
34 arm_compute::CLTensor m_OutputGather;
36 mutable arm_compute::CLPixelWiseMultiplication m_MulLayer;
37 mutable arm_compute::CLReductionOperation m_ReduceSumLayer;
38 mutable arm_compute::CLGather m_GatherLayer;
39 mutable arm_compute::CLReshapeLayer m_ReshapeLayer;
arm_compute::Status ClGatherNdWorkloadValidate(const TensorInfo ¶msInfo, const TensorInfo &indicesInfo, const TensorInfo &outputInfo)
Copyright (c) 2021 ARM Limited and Contributors.
ClGatherNdWorkload(const GatherNdQueueDescriptor &descriptor, const WorkloadInfo &info, const arm_compute::CLCompileContext &clCompileContext)
virtual void Execute() const override
Contains information about TensorInfos of a layer.