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/_neon_rank_workload_8hpp_source.xhtml | 126 ++++++++++++++++++++++++++++ 1 file changed, 126 insertions(+) create mode 100644 21.02/_neon_rank_workload_8hpp_source.xhtml (limited to '21.02/_neon_rank_workload_8hpp_source.xhtml') diff --git a/21.02/_neon_rank_workload_8hpp_source.xhtml b/21.02/_neon_rank_workload_8hpp_source.xhtml new file mode 100644 index 0000000000..839ade343c --- /dev/null +++ b/21.02/_neon_rank_workload_8hpp_source.xhtml @@ -0,0 +1,126 @@ + + + + + + + + + + + + + +ArmNN: src/backends/neon/workloads/NeonRankWorkload.hpp Source File + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  21.02 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+
NeonRankWorkload.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 "NeonWorkloadUtils.hpp"
12 
13 namespace armnn
14 {
15 
16 struct NeonRankWorkload : public BaseWorkload<RankQueueDescriptor>
17 {
18 public:
19  using BaseWorkload<RankQueueDescriptor>::BaseWorkload;
20  virtual void Execute() const override
21  {
22  const NeonTensorHandle* neonTensorHandle = PolymorphicDowncast<const NeonTensorHandle*>(m_Data.m_Inputs[0]);
23  const int32_t rank = static_cast<int32_t>(neonTensorHandle->GetShape().GetNumDimensions());
24 
25  std::memcpy(GetOutputTensorData<void>(0, m_Data), &rank, sizeof(int32_t));
26  m_Data.m_Outputs[0]->Unmap();
27  }
28 };
29 
30 } //namespace armnn
armnn::NeonTensorHandle::GetShape
TensorShape GetShape() const override
Get the number of elements for each dimension ordered from slowest iterating dimension to fastest ite...
Definition: NeonTensorHandle.hpp:94
+
WorkloadData.hpp
+
armnn::BaseWorkload< RankQueueDescriptor >::m_Data
const RankQueueDescriptor m_Data
Definition: Workload.hpp:46
+ +
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_software_tools.dox:6
+ +
NeonWorkloadUtils.hpp
+ +
Workload.hpp
+
armnn::QueueDescriptor::m_Outputs
std::vector< ITensorHandle * > m_Outputs
Definition: WorkloadData.hpp:31
+
armnn::TensorShape::GetNumDimensions
unsigned int GetNumDimensions() const
Function that returns the tensor rank.
Definition: Tensor.cpp:174
+
armnn::QueueDescriptor::m_Inputs
std::vector< ITensorHandle * > m_Inputs
Definition: WorkloadData.hpp:30
+
armnn::NeonRankWorkload::Execute
virtual void Execute() const override
Definition: NeonRankWorkload.hpp:20
+
+
+ + + + -- cgit v1.2.1