From 6940dd720ebb6b3d1df8ca203ab696daefe58189 Mon Sep 17 00:00:00 2001 From: Jim Flynn Date: Fri, 20 Mar 2020 12:25:56 +0000 Subject: renamed Documentation folder 20.02 and added .nojekyll file Signed-off-by: Jim Flynn --- 20.02/_neon_rsqrt_workload_8cpp_source.xhtml | 131 +++++++++++++++++++++++++++ 1 file changed, 131 insertions(+) create mode 100644 20.02/_neon_rsqrt_workload_8cpp_source.xhtml (limited to '20.02/_neon_rsqrt_workload_8cpp_source.xhtml') diff --git a/20.02/_neon_rsqrt_workload_8cpp_source.xhtml b/20.02/_neon_rsqrt_workload_8cpp_source.xhtml new file mode 100644 index 0000000000..a950e4c884 --- /dev/null +++ b/20.02/_neon_rsqrt_workload_8cpp_source.xhtml @@ -0,0 +1,131 @@ + + + + + + + + + + + + + +ArmNN: src/backends/neon/workloads/NeonRsqrtWorkload.cpp Source File + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  20.02 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+
NeonRsqrtWorkload.cpp
+
+
+Go to the documentation of this file.
1 //
2 // Copyright © 2019 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #include "NeonRsqrtWorkload.hpp"
7 
8 #include "NeonWorkloadUtils.hpp"
9 
10 #include <aclCommon/ArmComputeTensorHandle.hpp>
11 #include <aclCommon/ArmComputeTensorUtils.hpp>
12 
13 #include <boost/cast.hpp>
14 
15 namespace armnn
16 {
17 
18 arm_compute::Status NeonRsqrtWorkloadValidate(const TensorInfo& input, const TensorInfo& output)
19 {
20  const arm_compute::TensorInfo aclInput = armcomputetensorutils::BuildArmComputeTensorInfo(input);
21  const arm_compute::TensorInfo aclOutput = armcomputetensorutils::BuildArmComputeTensorInfo(output);
22 
23  return arm_compute::NERsqrtLayer::validate(&aclInput, &aclOutput);
24 }
25 
26 NeonRsqrtWorkload::NeonRsqrtWorkload(const RsqrtQueueDescriptor& descriptor, const WorkloadInfo& info)
27  : BaseWorkload<RsqrtQueueDescriptor>(descriptor, info)
28 {
29  m_Data.ValidateInputsOutputs("NeonRsqrtWorkload", 1, 1);
30 
31  arm_compute::ITensor& input = boost::polymorphic_downcast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
32  arm_compute::ITensor& output = boost::polymorphic_downcast<IAclTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();
33 
34  m_RsqrtLayer.configure(&input, &output);
35 }
36 
37 void NeonRsqrtWorkload::Execute() const
38 {
39  ARMNN_SCOPED_PROFILING_EVENT_NEON("NeonRsqrtWorkload_Execute");
40  m_RsqrtLayer.run();
41 }
42 
43 } // namespace armnn
armnn::NeonRsqrtWorkload::NeonRsqrtWorkload
NeonRsqrtWorkload(const RsqrtQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition: NeonRsqrtWorkload.cpp:26
+ +
armnn::BaseWorkload< RsqrtQueueDescriptor >::m_Data
const RsqrtQueueDescriptor m_Data
Definition: Workload.hpp:46
+
ARMNN_SCOPED_PROFILING_EVENT_NEON
#define ARMNN_SCOPED_PROFILING_EVENT_NEON(name)
Definition: NeonWorkloadUtils.hpp:17
+ +
ArmComputeTensorUtils.hpp
+
armnn::QueueDescriptor::ValidateInputsOutputs
void ValidateInputsOutputs(const std::string &descName, unsigned int numExpectedIn, unsigned int numExpectedOut) const
Definition: WorkloadData.cpp:461
+
armnn
Copyright (c) 2020 ARM Limited.
Definition: 00_introduction.dox:25
+ +
armnn::Status
Status
enumeration
Definition: Types.hpp:26
+
NeonWorkloadUtils.hpp
+
armnn::NeonRsqrtWorkload::Execute
virtual void Execute() const override
Definition: NeonRsqrtWorkload.cpp:37
+
armnn::NeonRsqrtWorkloadValidate
arm_compute::Status NeonRsqrtWorkloadValidate(const TensorInfo &input, const TensorInfo &output)
Definition: NeonRsqrtWorkload.cpp:18
+
armnn::QueueDescriptor::m_Outputs
std::vector< ITensorHandle * > m_Outputs
Definition: WorkloadData.hpp:31
+
NeonRsqrtWorkload.hpp
+
ArmComputeTensorHandle.hpp
+
armnn::WorkloadInfo
Contains information about inputs and outputs to a layer.
Definition: WorkloadInfo.hpp:16
+
armnn::QueueDescriptor::m_Inputs
std::vector< ITensorHandle * > m_Inputs
Definition: WorkloadData.hpp:30
+
+
+ + + + -- cgit v1.2.1