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_prelu_workload_8cpp_source.xhtml | 130 +++++++++++++++++++++++++++ 1 file changed, 130 insertions(+) create mode 100644 20.02/_neon_prelu_workload_8cpp_source.xhtml (limited to '20.02/_neon_prelu_workload_8cpp_source.xhtml') diff --git a/20.02/_neon_prelu_workload_8cpp_source.xhtml b/20.02/_neon_prelu_workload_8cpp_source.xhtml new file mode 100644 index 0000000000..fcee9af67c --- /dev/null +++ b/20.02/_neon_prelu_workload_8cpp_source.xhtml @@ -0,0 +1,130 @@ + + + + + + + + + + + + + +ArmNN: src/backends/neon/workloads/NeonPreluWorkload.cpp Source File + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  20.02 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+
NeonPreluWorkload.cpp
+
+
+Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #include "NeonPreluWorkload.hpp"
7 #include "NeonWorkloadUtils.hpp"
9 
10 #include <arm_compute/runtime/NEON/functions/NEPReluLayer.h>
11 
12 namespace armnn
13 {
14 
16  const TensorInfo& alpha,
17  const TensorInfo& output)
18 {
19  const arm_compute::TensorInfo aclInput = armcomputetensorutils::BuildArmComputeTensorInfo(input);
20  const arm_compute::TensorInfo aclAlpha = armcomputetensorutils::BuildArmComputeTensorInfo(alpha);
21  const arm_compute::TensorInfo aclOutput = armcomputetensorutils::BuildArmComputeTensorInfo(output);
22 
23  return arm_compute::NEPReluLayer::validate(&aclInput,
24  &aclAlpha,
25  &aclOutput);
26 }
27 
29  const WorkloadInfo& info)
30  : BaseWorkload<PreluQueueDescriptor>(descriptor, info)
31 {
32  m_Data.ValidateInputsOutputs("NeonPreluWorkload", 1, 1);
33 
34  arm_compute::ITensor& input = boost::polymorphic_downcast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
35  arm_compute::ITensor& alpha = boost::polymorphic_downcast<IAclTensorHandle*>(m_Data.m_Inputs[1])->GetTensor();
36  arm_compute::ITensor& output = boost::polymorphic_downcast<IAclTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();
37 
38  auto layer = std::make_unique<arm_compute::NEPReluLayer>();
39  layer->configure(&input, &alpha, &output);
40 
41  m_PreluLayer.reset(layer.release());
42 }
43 
45 {
46  ARMNN_SCOPED_PROFILING_EVENT_NEON("NeonPreluWorkload_Execute");
47  m_PreluLayer->run();
48 }
49 
50 } //namespace armnn
+ +
NeonPreluWorkload(const PreluQueueDescriptor &descriptor, const WorkloadInfo &info)
+
const PreluQueueDescriptor m_Data
Definition: Workload.hpp:46
+
#define ARMNN_SCOPED_PROFILING_EVENT_NEON(name)
+
void ValidateInputsOutputs(const std::string &descName, unsigned int numExpectedIn, unsigned int numExpectedOut) const
+
Copyright (c) 2020 ARM Limited.
+ + +
Status
enumeration
Definition: Types.hpp:26
+
void Execute() const override
+ + +
std::vector< ITensorHandle * > m_Outputs
+
Contains information about inputs and outputs to a layer.
+
std::vector< ITensorHandle * > m_Inputs
+
arm_compute::Status NeonPreluWorkloadValidate(const TensorInfo &input, const TensorInfo &alpha, const TensorInfo &output)
+
+
+ + + + -- cgit v1.2.1