patch/23.02/_cl_gather_workload_8cpp_source.xhtml

 //
 // Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
 // SPDX-License-Identifier: MIT
 //

 #include "ClGatherWorkload.hpp"
 #include "ClWorkloadUtils.hpp"
 #include <aclCommon/ArmComputeUtils.hpp>
 #include <cl/ClTensorHandle.hpp>

 using namespace armnn::armcomputetensorutils;

 namespace armnn
 {
 arm_compute::Status ClGatherWorkloadValidate(const TensorInfo& input,
                                              const TensorInfo& indices,
                                              const TensorInfo& output,
                                              const GatherDescriptor& descriptor)
 {
     const arm_compute::TensorInfo aclInput   = BuildArmComputeTensorInfo(input);
     const arm_compute::TensorInfo aclIndices = BuildArmComputeTensorInfo(indices);
     const arm_compute::TensorInfo aclOutput  = BuildArmComputeTensorInfo(output);

     int aclAxis = ComputeAclAxis(descriptor.m_Axis, input);

     return arm_compute::CLGather::validate(&aclInput, &aclIndices, &aclOutput, aclAxis);
 }

 ClGatherWorkload::ClGatherWorkload(const GatherQueueDescriptor& descriptor,
                                    const WorkloadInfo& info,
                                    const arm_compute::CLCompileContext& clCompileContext)
         : ClBaseWorkload<GatherQueueDescriptor>(descriptor, info)
 {
     // Report Profiling Details
     ARMNN_REPORT_PROFILING_WORKLOAD_DESC("ClGatherWorkload_Construct",
                                          descriptor.m_Parameters,
                                          info,
                                          this->GetGuid());

     m_Data.ValidateInputsOutputs("ClGatherWorkload", 2, 1);

     arm_compute::ICLTensor& input    = static_cast<IClTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
     arm_compute::ICLTensor& indices  = static_cast<IClTensorHandle*>(m_Data.m_Inputs[1])->GetTensor();
     arm_compute::ICLTensor& output   = static_cast<IClTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();

     int aclAxis = ComputeAclAxis(descriptor.m_Parameters.m_Axis, info.m_InputTensorInfos[0]);

     {
         ARMNN_SCOPED_PROFILING_EVENT(Compute::Undefined, "ClGatherWorkload_configure");
         m_Layer.configure(clCompileContext, &input, &indices, &output, aclAxis);
     }
 };

 void ClGatherWorkload::Execute() const
 {
     ARMNN_SCOPED_PROFILING_EVENT_CL_GUID("ClGatherWorkload_Execute", this->GetGuid());
     RunClFunction(m_Layer, CHECK_LOCATION());
 }
 } // namespace armnn
ARMNN_SCOPED_PROFILING_EVENT_CL_GUID
#define ARMNN_SCOPED_PROFILING_EVENT_CL_GUID(name, guid)
Definition: ClWorkloadUtils.hpp:28

-ClGatherWorkload.hpp

-armnn::ComputeAclAxis
int ComputeAclAxis(const int &armnnAxis, const armnn::TensorInfo &tensor)
Function to convert ArmNN axis (left to right) to ACL axis (right to left) ranging from [-rank...
Definition: ArmComputeUtils.hpp:264

-armnn::GatherQueueDescriptor
Definition: WorkloadData.hpp:502

-ArmComputeUtils.hpp

-armnn::TensorInfo
Definition: Tensor.hpp:152

-armnn::ClGatherWorkload::ClGatherWorkload
ClGatherWorkload(const GatherQueueDescriptor &descriptor, const WorkloadInfo &info, const arm_compute::CLCompileContext &clCompileContext)
Definition: ClGatherWorkload.cpp:29

-armnn::RunClFunction
void RunClFunction(arm_compute::IFunction &function, const CheckLocation &location)
Definition: ClWorkloadUtils.hpp:160

-armnn::BaseWorkload< GatherQueueDescriptor >::GetGuid
arm::pipe::ProfilingGuid GetGuid() const final
Definition: Workload.hpp:61

-armnn::QueueDescriptor::ValidateInputsOutputs
void ValidateInputsOutputs(const std::string &descName, unsigned int numExpectedIn, unsigned int numExpectedOut) const
Definition: WorkloadData.cpp:475

-armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6

-armnn::QueueDescriptorWithParameters::m_Parameters
LayerDescriptor m_Parameters
Definition: WorkloadData.hpp:66

-ARMNN_SCOPED_PROFILING_EVENT
#define ARMNN_SCOPED_PROFILING_EVENT(backendId, name)
Definition: Profiling.hpp:220

-armnn::WorkloadInfo::m_InputTensorInfos
std::vector< TensorInfo > m_InputTensorInfos
Definition: WorkloadInfo.hpp:18

-armnn::BaseWorkload< GatherQueueDescriptor >::m_Data
GatherQueueDescriptor m_Data
Definition: Workload.hpp:83

-armnn::Compute::Undefined

-armnn::GatherDescriptor
A GatherDescriptor for the GatherLayer.
Definition: Descriptors.hpp:912

-armnn::Status
Status
enumeration
Definition: Types.hpp:42

-armnn::ClGatherWorkloadValidate
arm_compute::Status ClGatherWorkloadValidate(const TensorInfo &input, const TensorInfo &indices, const TensorInfo &output, const GatherDescriptor &descriptor)
Definition: ClGatherWorkload.cpp:15

-armnn::ClBaseWorkload
Definition: ClBaseWorkload.hpp:13

-CHECK_LOCATION
#define CHECK_LOCATION()
Definition: Exceptions.hpp:203

-armnn::GatherDescriptor::m_Axis
int32_t m_Axis
The axis in params to gather indices from.
Definition: Descriptors.hpp:928

-ClTensorHandle.hpp

-armnn::QueueDescriptor::m_Outputs
std::vector< ITensorHandle * > m_Outputs
Definition: WorkloadData.hpp:27

-armnn::BoostLogSeverityMapping::info

-armnn::IClTensorHandle
Definition: IClTensorHandle.hpp:13

-ARMNN_REPORT_PROFILING_WORKLOAD_DESC
#define ARMNN_REPORT_PROFILING_WORKLOAD_DESC(name, desc, infos, guid)
Definition: Profiling.hpp:227

-armnn::WorkloadInfo
Contains information about TensorInfos of a layer.
Definition: WorkloadInfo.hpp:16

-armnn::QueueDescriptor::m_Inputs
std::vector< ITensorHandle * > m_Inputs
Definition: WorkloadData.hpp:26

-armnn::armcomputetensorutils
Definition: ArmComputeTensorUtils.cpp:16

-ClWorkloadUtils.hpp

-armnn::ClGatherWorkload::Execute
void Execute() const override
Definition: ClGatherWorkload.cpp:54

+Go to the documentation of this file. //

+ // Copyright © 2020 Arm Ltd and Contributors. All rights reserved.

+ // SPDX-License-Identifier: MIT

+ //

+

+ #include "ClGatherWorkload.hpp"

+ #include "ClWorkloadUtils.hpp"

+ #include <aclCommon/ArmComputeUtils.hpp>

+ #include <cl/ClTensorHandle.hpp>

+

+ using namespace armnn::armcomputetensorutils;

+

+ namespace armnn

+ {

+ arm_compute::Status ClGatherWorkloadValidate(const TensorInfo& input,

+                                              const TensorInfo& indices,

+                                              const TensorInfo& output,

+                                              const GatherDescriptor& descriptor)

+ {

+     const arm_compute::TensorInfo aclInput   = BuildArmComputeTensorInfo(input);

+     const arm_compute::TensorInfo aclIndices = BuildArmComputeTensorInfo(indices);

+     const arm_compute::TensorInfo aclOutput  = BuildArmComputeTensorInfo(output);

+

+     int aclAxis = ComputeAclAxis(descriptor.m_Axis, input);

+

+     return arm_compute::CLGather::validate(&aclInput, &aclIndices, &aclOutput, aclAxis);

+ }

+

+ ClGatherWorkload::ClGatherWorkload(const GatherQueueDescriptor& descriptor,

+                                    const WorkloadInfo& info,

+                                    const arm_compute::CLCompileContext& clCompileContext)

+         : ClBaseWorkload<GatherQueueDescriptor>(descriptor, info)

+ {

+     // Report Profiling Details

+     ARMNN_REPORT_PROFILING_WORKLOAD_DESC("ClGatherWorkload_Construct",

+                                          descriptor.m_Parameters,

+                                          info,

+                                          this->GetGuid());

+

+     m_Data.ValidateInputsOutputs("ClGatherWorkload", 2, 1);

+

+     arm_compute::ICLTensor& input    = static_cast<IClTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();

+     arm_compute::ICLTensor& indices  = static_cast<IClTensorHandle*>(m_Data.m_Inputs[1])->GetTensor();

+     arm_compute::ICLTensor& output   = static_cast<IClTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();

+

+     int aclAxis = ComputeAclAxis(descriptor.m_Parameters.m_Axis, info.m_InputTensorInfos[0]);

+

+     {

+         ARMNN_SCOPED_PROFILING_EVENT(Compute::Undefined, "ClGatherWorkload_configure");

+         m_Layer.configure(clCompileContext, &input, &indices, &output, aclAxis);

+     }

+ };

+

+ void ClGatherWorkload::Execute() const

+ {

+     ARMNN_SCOPED_PROFILING_EVENT_CL_GUID("ClGatherWorkload_Execute", this->GetGuid());

+     RunClFunction(m_Layer, CHECK_LOCATION());

+ }

+ } // namespace armnn