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_slice_workload_8cpp_source.xhtml | 137 +++++++++++++++++++++++++++ 1 file changed, 137 insertions(+) create mode 100644 20.02/_neon_slice_workload_8cpp_source.xhtml (limited to '20.02/_neon_slice_workload_8cpp_source.xhtml') diff --git a/20.02/_neon_slice_workload_8cpp_source.xhtml b/20.02/_neon_slice_workload_8cpp_source.xhtml new file mode 100644 index 0000000000..3b2efec3f6 --- /dev/null +++ b/20.02/_neon_slice_workload_8cpp_source.xhtml @@ -0,0 +1,137 @@ + + + + + + + + + + + + + +ArmNN: src/backends/neon/workloads/NeonSliceWorkload.cpp Source File + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  20.02 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+
NeonSliceWorkload.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 "NeonSliceWorkload.hpp"
7 
8 #include "NeonWorkloadUtils.hpp"
9 
10 #include <aclCommon/ArmComputeTensorUtils.hpp>
11 
12 #include <neon/NeonTensorHandle.hpp>
13 
14 #include <arm_compute/core/Error.h>
15 
16 namespace armnn
17 {
18 
19 arm_compute::Status NeonSliceWorkloadValidate(const TensorInfo& input,
20  const TensorInfo& output,
21  const SliceDescriptor& descriptor)
22 {
23  const arm_compute::TensorInfo aclInputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(input);
24  const arm_compute::TensorInfo aclOutputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(output);
25 
26  arm_compute::Coordinates starts;
27  arm_compute::Coordinates ends;
28 
29  std::tie(starts, ends) = SetNeonSliceData(descriptor.m_Begin, descriptor.m_Size);
30 
31  return arm_compute::NESlice::validate(&aclInputInfo, &aclOutputInfo, starts, ends);
32 }
33 
34 NeonSliceWorkload::NeonSliceWorkload(const SliceQueueDescriptor& descriptor,
35  const WorkloadInfo& info)
36  : BaseWorkload<SliceQueueDescriptor>(descriptor, info)
37 {
38  m_Data.ValidateInputsOutputs("NeonSliceWorkload", 1, 1);
39 
40  arm_compute::ITensor& input = boost::polymorphic_downcast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
41  arm_compute::ITensor& output = boost::polymorphic_downcast<IAclTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();
42 
43  arm_compute::Coordinates starts;
44  arm_compute::Coordinates ends;
45 
46  std::tie(starts, ends) = SetNeonSliceData(m_Data.m_Parameters.m_Begin, m_Data.m_Parameters.m_Size);
47 
48  m_SliceFunction.configure(&input, &output, starts, ends);
49 }
50 
51 void NeonSliceWorkload::Execute() const
52 {
53  ARMNN_SCOPED_PROFILING_EVENT_NEON("NeonSliceWorkload_Execute");
54  m_SliceFunction.run();
55 }
56 
57 } // namespace armnn
+
armnn::Coordinates
std::array< unsigned int, MaxNumOfTensorDimensions > Coordinates
Definition: InternalTypes.hpp:80
+
armnn::BaseWorkload< SliceQueueDescriptor >::m_Data
const SliceQueueDescriptor m_Data
Definition: Workload.hpp:46
+
ARMNN_SCOPED_PROFILING_EVENT_NEON
#define ARMNN_SCOPED_PROFILING_EVENT_NEON(name)
Definition: NeonWorkloadUtils.hpp:17
+
armnn::SliceDescriptor::m_Size
std::vector< unsigned int > m_Size
Size of the slice in each dimension.
Definition: Descriptors.hpp:946
+
NeonTensorHandle.hpp
+
ArmComputeTensorUtils.hpp
+
armnn::NeonSliceWorkloadValidate
arm_compute::Status NeonSliceWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const SliceDescriptor &descriptor)
Definition: NeonSliceWorkload.cpp:19
+
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::SliceDescriptor::m_Begin
std::vector< unsigned int > m_Begin
Beginning indices of the slice in each dimension.
Definition: Descriptors.hpp:943
+
armnn::SetNeonSliceData
auto SetNeonSliceData(const std::vector< unsigned int > &m_begin, const std::vector< unsigned int > &m_size)
Definition: NeonWorkloadUtils.hpp:88
+
armnn::Status
Status
enumeration
Definition: Types.hpp:26
+
NeonWorkloadUtils.hpp
+
armnn::NeonSliceWorkload::Execute
virtual void Execute() const override
Definition: NeonSliceWorkload.cpp:51
+
NeonSliceWorkload.hpp
+
armnn::SliceDescriptor
A SliceDescriptor for the SliceLayer.
Definition: Descriptors.hpp:927
+
armnn::NeonSliceWorkload::NeonSliceWorkload
NeonSliceWorkload(const SliceQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition: NeonSliceWorkload.cpp:34
+ +
armnn::QueueDescriptor::m_Outputs
std::vector< ITensorHandle * > m_Outputs
Definition: WorkloadData.hpp:31
+
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