From fd627ffaec8fd8801d980b4c91ee7c0607ab6aaf Mon Sep 17 00:00:00 2001 From: Jan Eilers Date: Thu, 25 Feb 2021 17:44:00 +0000 Subject: IVGCVSW-5687 Update Doxygen Docu * Update Doxygen Documentation for 21.02 release Signed-off-by: Jan Eilers Change-Id: I9ed2f9caab038836ea99d7b378d7899fe431a4e5 --- 21.02/classarmnn_1_1_ref_gather_workload.xhtml | 204 +++++++++++++++++++++++++ 1 file changed, 204 insertions(+) create mode 100644 21.02/classarmnn_1_1_ref_gather_workload.xhtml (limited to '21.02/classarmnn_1_1_ref_gather_workload.xhtml') diff --git a/21.02/classarmnn_1_1_ref_gather_workload.xhtml b/21.02/classarmnn_1_1_ref_gather_workload.xhtml new file mode 100644 index 0000000000..a1c9be71f9 --- /dev/null +++ b/21.02/classarmnn_1_1_ref_gather_workload.xhtml @@ -0,0 +1,204 @@ + + + + + + + + + + + + + +ArmNN: RefGatherWorkload Class Reference + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  21.02 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+ +
+
RefGatherWorkload Class Reference
+
+
+ +

#include <RefGatherWorkload.hpp>

+
+Inheritance diagram for RefGatherWorkload:
+
+
+ + +BaseWorkload< GatherQueueDescriptor > +IWorkload + +
+ + + + + + + + + + + + + + + + + + +

+Public Member Functions

void Execute () const override
 
- Public Member Functions inherited from BaseWorkload< GatherQueueDescriptor >
 BaseWorkload (const GatherQueueDescriptor &descriptor, const WorkloadInfo &info)
 
void PostAllocationConfigure () override
 
const GatherQueueDescriptorGetData () const
 
profiling::ProfilingGuid GetGuid () const final
 
- Public Member Functions inherited from IWorkload
virtual ~IWorkload ()
 
virtual void RegisterDebugCallback (const DebugCallbackFunction &)
 
+ + + + + + +

+Additional Inherited Members

- Protected Attributes inherited from BaseWorkload< GatherQueueDescriptor >
const GatherQueueDescriptor m_Data
 
const profiling::ProfilingGuid m_Guid
 
+

Detailed Description

+
+

Definition at line 19 of file RefGatherWorkload.hpp.

+

Member Function Documentation

+ +

◆ Execute()

+ +
+
+ + + + + +
+ + + + + + + +
void Execute () const
+
+overridevirtual
+
+ +

Implements IWorkload.

+ +

Definition at line 16 of file RefGatherWorkload.cpp.

+ +

References ARMNN_SCOPED_PROFILING_EVENT, armnn::CpuRef, armnn::Gather, armnn::GetTensorInfo(), GatherDescriptor::m_Axis, BaseWorkload< GatherQueueDescriptor >::m_Data, QueueDescriptor::m_Inputs, QueueDescriptor::m_Outputs, and QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters.

+
17 {
18  ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefGatherWorkload_Execute");
19 
20  const TensorInfo& inputInfo0 = GetTensorInfo(m_Data.m_Inputs[0]);
21  const TensorInfo& inputInfo1 = GetTensorInfo(m_Data.m_Inputs[1]);
22  const TensorInfo& outputInfo = GetTensorInfo(m_Data.m_Outputs[0]);
23 
24  std::unique_ptr<Decoder<float>> decoderPtr = MakeDecoder<float>(inputInfo0, m_Data.m_Inputs[0]->Map());
25  Decoder<float>& decoder = *decoderPtr;
26 
27  const int32_t* indicesData = GetInputTensorData<int32_t>(1, m_Data);
28 
29  std::unique_ptr<Encoder<float>> encoderPtr = MakeEncoder<float>(outputInfo, m_Data.m_Outputs[0]->Map());
30  Encoder<float>& encoder = *encoderPtr;
31 
32  Gather(inputInfo0, inputInfo1, outputInfo, decoder, indicesData, encoder, m_Data.m_Parameters.m_Axis);
33 }
CPU Execution: Reference C++ kernels.
+
const GatherQueueDescriptor m_Data
Definition: Workload.hpp:46
+ +
#define ARMNN_SCOPED_PROFILING_EVENT(backendId, name)
Definition: Profiling.hpp:173
+ +
int32_t m_Axis
The axis in params to gather indices from.
+
std::vector< ITensorHandle * > m_Outputs
+
std::vector< ITensorHandle * > m_Inputs
+
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers
+
+
+
+
The documentation for this class was generated from the following files: +
+
+ + + + -- cgit v1.2.1