From add078b7f9a658d52ee6e8fe6771ea0517c07674 Mon Sep 17 00:00:00 2001 From: mathad01 Date: Mon, 10 May 2021 16:51:20 +0100 Subject: IVGCVSW-5908 Update 21.05 Doxygen Documents Signed-off-by: mathad01 Change-Id: I95316d4fc5f9d10185492dc835bb2411c1daea7b --- 21.05/_ref_rank_workload_8hpp_source.xhtml | 130 +++++++++++++++++++++++++++++ 1 file changed, 130 insertions(+) create mode 100644 21.05/_ref_rank_workload_8hpp_source.xhtml (limited to '21.05/_ref_rank_workload_8hpp_source.xhtml') diff --git a/21.05/_ref_rank_workload_8hpp_source.xhtml b/21.05/_ref_rank_workload_8hpp_source.xhtml new file mode 100644 index 0000000000..9efe05c1d8 --- /dev/null +++ b/21.05/_ref_rank_workload_8hpp_source.xhtml @@ -0,0 +1,130 @@ + + + + + + + + + + + + + +ArmNN: src/backends/reference/workloads/RefRankWorkload.hpp Source File + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  21.05 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+
RefRankWorkload.hpp
+
+
+Go to the documentation of this file.
1 //
2 // Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
8 #include <backendsCommon/Workload.hpp>
9 #include <backendsCommon/WorkloadData.hpp>
10 
11 #include "RefWorkloadUtils.hpp"
12 
13 namespace armnn
14 {
15 
16 struct RefRankWorkload : public BaseWorkload<RankQueueDescriptor>
17 {
18 public:
19  using BaseWorkload<RankQueueDescriptor>::BaseWorkload;
20  virtual void Execute() const override
21  {
22  Execute(m_Data.m_Inputs, m_Data.m_Outputs);
23 
24  }
25  void ExecuteAsync(WorkingMemDescriptor& workingMemDescriptor) override
26  {
27  Execute(workingMemDescriptor.m_Inputs, workingMemDescriptor.m_Outputs);
28  }
29 
30 private:
31  void Execute(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs) const
32  {
33  const int32_t rank = static_cast<int32_t>(GetTensorInfo(inputs[0]).GetNumDimensions());
34 
35  std::memcpy(outputs[0]->Map(), &rank, sizeof(int32_t));
36  outputs[0]->Unmap();
37  }
38 };
39 
40 } //namespace armnn
41 
42 
43 
44 
WorkloadData.hpp
+
RefWorkloadUtils.hpp
+
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_software_tools.dox:6
+ +
armnn::RefRankWorkload::ExecuteAsync
void ExecuteAsync(WorkingMemDescriptor &workingMemDescriptor) override
Definition: RefRankWorkload.hpp:25
+ + + +
armnn::RefRankWorkload::Execute
virtual void Execute() const override
Definition: RefRankWorkload.hpp:20
+ +
Workload.hpp
+ +
armnn::QueueDescriptor::m_Outputs
std::vector< ITensorHandle * > m_Outputs
Definition: WorkloadData.hpp:31
+ +
armnn::QueueDescriptor::m_Inputs
std::vector< ITensorHandle * > m_Inputs
Definition: WorkloadData.hpp:30
+
armnn::GetTensorInfo
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers
Definition: RefWorkloadUtils.hpp:26
+
armnn::TensorInfo::GetNumDimensions
unsigned int GetNumDimensions() const
Definition: Tensor.hpp:191
+
+
+ + + + -- cgit v1.2.1