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/classarmnn_1_1_neon_splitter_workload.xhtml | 236 ++++++++++++++++++++++ 1 file changed, 236 insertions(+) create mode 100644 20.02/classarmnn_1_1_neon_splitter_workload.xhtml (limited to '20.02/classarmnn_1_1_neon_splitter_workload.xhtml') diff --git a/20.02/classarmnn_1_1_neon_splitter_workload.xhtml b/20.02/classarmnn_1_1_neon_splitter_workload.xhtml new file mode 100644 index 0000000000..574958a98b --- /dev/null +++ b/20.02/classarmnn_1_1_neon_splitter_workload.xhtml @@ -0,0 +1,236 @@ + + + + + + + + + + + + + +ArmNN: NeonSplitterWorkload Class Reference + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  20.02 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+Public Member Functions | +List of all members
+
+
NeonSplitterWorkload Class Reference
+
+
+ +

#include <NeonSplitterWorkload.hpp>

+
+Inheritance diagram for NeonSplitterWorkload:
+
+
+ + +BaseWorkload< SplitterQueueDescriptor > +IWorkload + +
+ + + + + + + + + + + + + + + + + + + + +

+Public Member Functions

 NeonSplitterWorkload (const SplitterQueueDescriptor &descriptor, const WorkloadInfo &info)
 
void Execute () const override
 
- Public Member Functions inherited from BaseWorkload< SplitterQueueDescriptor >
 BaseWorkload (const SplitterQueueDescriptor &descriptor, const WorkloadInfo &info)
 
void PostAllocationConfigure () override
 
const SplitterQueueDescriptor & GetData () const
 
profiling::ProfilingGuid GetGuid () const final
 
- Public Member Functions inherited from IWorkload
virtual ~IWorkload ()
 
virtual void RegisterDebugCallback (const DebugCallbackFunction &)
 
+ + + + + + +

+Additional Inherited Members

- Protected Attributes inherited from BaseWorkload< SplitterQueueDescriptor >
const SplitterQueueDescriptor m_Data
 
const profiling::ProfilingGuid m_Guid
 
+

Detailed Description

+
+

Definition at line 22 of file NeonSplitterWorkload.hpp.

+

Constructor & Destructor Documentation

+ +

◆ NeonSplitterWorkload()

+ +
+
+ + + + + + + + + + + + + + + + + + +
NeonSplitterWorkload (const SplitterQueueDescriptor & descriptor,
const WorkloadInfo & info 
)
+
+ +

Definition at line 55 of file NeonSplitterWorkload.cpp.

+ +

References armnn::ComputeSplitAxis(), ViewsDescriptor::GetNumDimensions(), BaseWorkload< SplitterQueueDescriptor >::m_Data, QueueDescriptor::m_Inputs, QueueDescriptor::m_Outputs, and QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters.

+
56  : BaseWorkload<SplitterQueueDescriptor>(descriptor, info)
57 {
58  bool allOutputsAreSubtensors = true;
59 
60  // Check that all outputs are sub-tensors
61  for (auto output : m_Data.m_Outputs)
62  {
63  if (output && !output->GetParent())
64  {
65  // Non sub-tensor input found so we need to execute the split function
66  allOutputsAreSubtensors = false;
67  break;
68  }
69  }
70 
71  if (allOutputsAreSubtensors)
72  {
73  // Can skip configuring the split function since it's not executed
74  return;
75  }
76 
77  arm_compute::ITensor& input = boost::polymorphic_downcast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
78 
79  std::vector<arm_compute::ITensor *> aclOutputs;
80  for (auto output : m_Data.m_Outputs)
81  {
82  arm_compute::ITensor& aclOutput = boost::polymorphic_pointer_downcast<IAclTensorHandle>(output)->GetTensor();
83  aclOutputs.emplace_back(&aclOutput);
84  }
85 
86  // Create the layer function
87  std::unique_ptr<arm_compute::NESplit> layer(new arm_compute::NESplit());
88 
89  // Configure input and output tensors
90  std::set<unsigned int> splitAxis = ComputeSplitAxis(descriptor.m_Parameters, m_Data.m_Inputs[0]->GetShape());
91  if (splitAxis.size() != 1)
92  {
93  throw InvalidArgumentException("Cannot derive split axis from SplitterDescriptor");
94  }
95 
96  unsigned int aclAxis = CalcAclAxis(descriptor.m_Parameters.GetNumDimensions(), *splitAxis.begin());
97  layer->configure(&input, aclOutputs, aclAxis);
98 
99  // Prepare
100  layer->prepare();
101  m_Layer.reset(layer.release());
102 }
armnn::BaseWorkload< SplitterQueueDescriptor >::m_Data
const SplitterQueueDescriptor m_Data
Definition: Workload.hpp:46
+
armnn::ComputeSplitAxis
std::set< unsigned int > ComputeSplitAxis(const armnn::SplitterDescriptor &desc, const TensorShape &input)
Definition: ArmComputeUtils.hpp:155
+
armnn::QueueDescriptor::m_Outputs
std::vector< ITensorHandle * > m_Outputs
Definition: WorkloadData.hpp:31
+ +
armnn::QueueDescriptor::m_Inputs
std::vector< ITensorHandle * > m_Inputs
Definition: WorkloadData.hpp:30
+
+
+
+

Member Function Documentation

+ +

◆ Execute()

+ +
+
+ + + + + +
+ + + + + + + +
void Execute () const
+ 		+overridevirtual
+
+ +

Implements IWorkload.

+ +

Definition at line 104 of file NeonSplitterWorkload.cpp.

+ +

References ARMNN_SCOPED_PROFILING_EVENT_NEON.

+
105 {
106  if (m_Layer)
107  {
108  ARMNN_SCOPED_PROFILING_EVENT_NEON("NeonSplitterWorkload_Execute");
109  m_Layer->run();
110  }
111 }
ARMNN_SCOPED_PROFILING_EVENT_NEON
#define ARMNN_SCOPED_PROFILING_EVENT_NEON(name)
Definition: NeonWorkloadUtils.hpp:17
+
+
+
+
The documentation for this class was generated from the following files: +
+
+ + + + -- cgit v1.2.1