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_transpose_workload_8cpp_source.xhtml | 136 +++++++++++++++++++++++ 1 file changed, 136 insertions(+) create mode 100644 20.02/_neon_transpose_workload_8cpp_source.xhtml (limited to '20.02/_neon_transpose_workload_8cpp_source.xhtml') diff --git a/20.02/_neon_transpose_workload_8cpp_source.xhtml b/20.02/_neon_transpose_workload_8cpp_source.xhtml new file mode 100644 index 0000000000..e58ff0cc07 --- /dev/null +++ b/20.02/_neon_transpose_workload_8cpp_source.xhtml @@ -0,0 +1,136 @@ + + + + + + + + + + + + + +ArmNN: src/backends/neon/workloads/NeonTransposeWorkload.cpp Source File + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  20.02 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+
NeonTransposeWorkload.cpp
+
+
+Go to the documentation of this file.
1 //
2 // Copyright © 2020 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #include "NeonTransposeWorkload.hpp"
7 #include <neon/NeonTensorHandle.hpp>
8 #include <aclCommon/ArmComputeTensorUtils.hpp>
9 
10 #include <arm_compute/core/Error.h>
11 
12 namespace armnn
13 {
14 
15 arm_compute::Status NeonTransposeWorkloadValidate(const TensorInfo& input,
16  const TensorInfo& output,
17  const TransposeDescriptor& descriptor)
18 {
19  const arm_compute::TensorInfo aclInputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(input);
20  const arm_compute::TensorInfo aclOutputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(output);
21  const armnn::PermutationVector& mappings = descriptor.m_DimMappings;
22 
23  return arm_compute::NEPermute::validate(&aclInputInfo, &aclOutputInfo,
24  armcomputetensorutils::BuildArmComputeTransposeVector(mappings));
25 }
26 
27 NeonTransposeWorkload::NeonTransposeWorkload(const TransposeQueueDescriptor& descriptor,
28  const WorkloadInfo& info)
29  : BaseWorkload<TransposeQueueDescriptor>(descriptor, info)
30 {
31  m_Data.ValidateInputsOutputs(GetName(), 1, 1);
32 
33  const arm_compute::ITensor& input = static_cast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
34  arm_compute::ITensor& output = static_cast<IAclTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();
35  const armnn::PermutationVector& mappings = m_Data.m_Parameters.m_DimMappings;
36 
37  // Run the layer.
38  m_PermuteFunction.configure(&input, &output,
39  armcomputetensorutils::BuildArmComputeTransposeVector(mappings));
40 }
41 
42 void NeonTransposeWorkload::Execute() const
43 {
44  ARMNN_SCOPED_PROFILING_EVENT_NEON(GetName() + "_Execute");
45  m_PermuteFunction.run();
46 }
47 
48 } // namespace armnn
+
armnn::BaseWorkload< TransposeQueueDescriptor >::m_Data
const TransposeQueueDescriptor m_Data
Definition: Workload.hpp:46
+
ARMNN_SCOPED_PROFILING_EVENT_NEON
#define ARMNN_SCOPED_PROFILING_EVENT_NEON(name)
Definition: NeonWorkloadUtils.hpp:17
+
NeonTensorHandle.hpp
+
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
+ +
armnn::NeonTransposeWorkload::NeonTransposeWorkload
NeonTransposeWorkload(const TransposeQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition: NeonTransposeWorkload.cpp:27
+
NeonTransposeWorkload.hpp
+ +
armnn::QueueDescriptor::m_Outputs
std::vector< ITensorHandle * > m_Outputs
Definition: WorkloadData.hpp:31
+
armnn::NeonTransposeWorkloadValidate
arm_compute::Status NeonTransposeWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const TransposeDescriptor &descriptor)
Definition: NeonTransposeWorkload.cpp:15
+
armnn::TransposeDescriptor
A TransposeDescriptor for the TransposeLayer.
Definition: Descriptors.hpp:1123
+ +
armnn::WorkloadInfo
Contains information about inputs and outputs to a layer.
Definition: WorkloadInfo.hpp:16
+
armnn::NeonTransposeWorkload::GetName
static const std::string & GetName()
Definition: NeonTransposeWorkload.hpp:25
+
armnn::QueueDescriptor::m_Inputs
std::vector< ITensorHandle * > m_Inputs
Definition: WorkloadData.hpp:30
+
armnn::TransposeDescriptor::m_DimMappings
PermutationVector m_DimMappings
Indicates how to translate tensor elements from a given source into the target destination, when source and target potentially have different memory layouts e.g.
Definition: Descriptors.hpp:1140
+
+
+ + + + -- cgit v1.2.1