ArmNN
 22.11
Public Member Functions | Static Public Member Functions | List of all members
NeonWorkloadFactory Class Reference

#include <NeonWorkloadFactory.hpp>

Inheritance diagram for NeonWorkloadFactory:
WorkloadFactoryBase IWorkloadFactory

Public Member Functions

 NeonWorkloadFactory (const std::shared_ptr< NeonMemoryManager > &memoryManager)
 
 NeonWorkloadFactory (const std::shared_ptr< NeonMemoryManager > &memoryManager, const IBackendInternal::IBackendSpecificModelContextPtr &modelContextPtr)
 
const BackendId & GetBackendId () const override
 
bool SupportsSubTensors () const override
 
std::unique_ptr< ITensorHandle > CreateSubTensorHandle (ITensorHandle &parent, TensorShape const &subTensorShape, unsigned int const *subTensorOrigin) const override
 
std::unique_ptr< ITensorHandle > CreateTensorHandle (const TensorInfo &tensorInfo, const bool IsMemoryManaged=true) const override
 
std::unique_ptr< ITensorHandle > CreateTensorHandle (const TensorInfo &tensorInfo, DataLayout dataLayout, const bool IsMemoryManaged=true) const override
 
std::unique_ptr< IWorkload > CreateWorkload (LayerType type, const QueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
- Public Member Functions inherited from WorkloadFactoryBase
std::unique_ptr< IWorkload > CreateInput (const InputQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateActivation (const ActivationQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateAddition (const AdditionQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateArgMinMax (const ArgMinMaxQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateBatchNormalization (const BatchNormalizationQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateBatchToSpaceNd (const BatchToSpaceNdQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateComparison (const ComparisonQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateConcat (const ConcatQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateConstant (const ConstantQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateConvertFp16ToFp32 (const ConvertFp16ToFp32QueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateConvertFp32ToFp16 (const ConvertFp32ToFp16QueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateConvolution2d (const Convolution2dQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateDebug (const DebugQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateDepthToSpace (const DepthToSpaceQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateDepthwiseConvolution2d (const DepthwiseConvolution2dQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateDequantize (const DequantizeQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateDetectionPostProcess (const DetectionPostProcessQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateDivision (const DivisionQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateElementwiseUnary (const ElementwiseUnaryQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateFakeQuantization (const FakeQuantizationQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateFloor (const FloorQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateFullyConnected (const FullyConnectedQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateGather (const GatherQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateInstanceNormalization (const InstanceNormalizationQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateL2Normalization (const L2NormalizationQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateLogSoftmax (const LogSoftmaxQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateLstm (const LstmQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateMaximum (const MaximumQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateMean (const MeanQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateMemCopy (const MemCopyQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateMemImport (const MemImportQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateMerge (const MergeQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateMinimum (const MinimumQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateMultiplication (const MultiplicationQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateNormalization (const NormalizationQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateOutput (const OutputQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreatePad (const PadQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreatePermute (const PermuteQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreatePooling2d (const Pooling2dQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreatePooling3d (const Pooling3dQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreatePreCompiled (const PreCompiledQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreatePrelu (const PreluQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateQuantize (const QuantizeQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateQuantizedLstm (const QuantizedLstmQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateRank (const RankQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateReshape (const ReshapeQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateResize (const ResizeQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateSlice (const SliceQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateSoftmax (const SoftmaxQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateSpaceToBatchNd (const SpaceToBatchNdQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateSpaceToDepth (const SpaceToDepthQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateSubtraction (const SubtractionQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateSplitter (const SplitterQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateStack (const StackQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateStridedSlice (const StridedSliceQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateSwitch (const SwitchQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateTranspose (const TransposeQueueDescriptor &, const WorkloadInfo &) const override
 
std::unique_ptr< IWorkload > CreateTransposeConvolution2d (const TransposeConvolution2dQueueDescriptor &, const WorkloadInfo &) const override
 
- Public Member Functions inherited from IWorkloadFactory
virtual ~IWorkloadFactory ()
 
virtual void AfterWorkloadsCreated ()
 

Static Public Member Functions

static bool IsLayerSupported (const Layer &layer, Optional< DataType > dataType, std::string &outReasonIfUnsupported)
 
static bool IsLayerSupported (const IConnectableLayer &layer, Optional< DataType > dataType, std::string &outReasonIfUnsupported, const ModelOptions &modelOptions)
 
- Static Public Member Functions inherited from IWorkloadFactory
static bool IsLayerSupported (const BackendId &backendId, const IConnectableLayer &layer, Optional< DataType > dataType, std::string &outReasonIfUnsupported)
 
static bool IsLayerSupported (const IConnectableLayer &layer, Optional< DataType > dataType, std::string &outReasonIfUnsupported)
 
static bool IsLayerSupported (const IConnectableLayer &layer, Optional< DataType > dataType, std::string &outReasonIfUnsupported, const ModelOptions &modelOptions)
 
static bool IsLayerSupported (const BackendId &backendId, const IConnectableLayer &layer, Optional< DataType > dataType, std::string &outReasonIfUnsupported, const ModelOptions &modelOptions)
 

Detailed Description

Definition at line 20 of file NeonWorkloadFactory.hpp.

Constructor & Destructor Documentation

◆ NeonWorkloadFactory() [1/2]

NeonWorkloadFactory ( const std::shared_ptr< NeonMemoryManager > &  memoryManager)

Definition at line 73 of file NeonWorkloadFactory.cpp.

74  : m_MemoryManager(memoryManager), m_ModelContextPtr(IBackendInternal::IBackendSpecificModelContextPtr{})
75 {
76  SetNumberOfThreads();
77 }
armnn::IBackendInternal::IBackendSpecificModelContextPtr
std::shared_ptr< IBackendModelContext > IBackendSpecificModelContextPtr
Definition: IBackendInternal.hpp:96

◆ NeonWorkloadFactory() [2/2]

NeonWorkloadFactory ( const std::shared_ptr< NeonMemoryManager > &  memoryManager,
const IBackendInternal::IBackendSpecificModelContextPtr &  modelContextPtr 
)

Definition at line 79 of file NeonWorkloadFactory.cpp.

81  : m_MemoryManager(memoryManager), m_ModelContextPtr(modelContextPtr)
82 {
83  SetNumberOfThreads();
84 }

Member Function Documentation

◆ CreateSubTensorHandle()

std::unique_ptr< ITensorHandle > CreateSubTensorHandle ( ITensorHandle &  parent,
TensorShape const &  subTensorShape,
unsigned int const *  subTensorOrigin 
) const
overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 86 of file NeonWorkloadFactory.cpp.

Referenced by NeonWorkloadFactory::SupportsSubTensors().

89 {
90  const arm_compute::TensorShape shape = armcomputetensorutils::BuildArmComputeTensorShape(subTensorShape);
91 
92  arm_compute::Coordinates coords;
93  coords.set_num_dimensions(subTensorShape.GetNumDimensions());
94  for (unsigned int i = 0; i < subTensorShape.GetNumDimensions(); i++)
95  {
96  // Arm compute indexes tensor coords in reverse order.
97  unsigned int revertedIndex = subTensorShape.GetNumDimensions() - i - 1;
98  coords.set(i, armnn::numeric_cast<int>(subTensorOrigin[revertedIndex]));
99  }
100 
101  const arm_compute::TensorShape parentShape = armcomputetensorutils::BuildArmComputeTensorShape(parent.GetShape());
102  if (!::arm_compute::error_on_invalid_subtensor(__func__, __FILE__, __LINE__, parentShape, coords, shape))
103  {
104  return nullptr;
105  }
106 
107  return std::make_unique<NeonSubTensorHandle>(
108  PolymorphicDowncast<IAclTensorHandle*>(&parent), shape, coords);
109 }
armnn::Coordinates
std::array< unsigned int, MaxNumOfTensorDimensions > Coordinates
Definition: InternalTypes.hpp:15

◆ CreateTensorHandle() [1/2]

std::unique_ptr< ITensorHandle > CreateTensorHandle ( const TensorInfo &  tensorInfo,
const bool  IsMemoryManaged = true 
) const
overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 111 of file NeonWorkloadFactory.cpp.

Referenced by NeonWorkloadFactory::SupportsSubTensors().

113 {
114  auto tensorHandle = std::make_unique<NeonTensorHandle>(tensorInfo);
115  if (IsMemoryManaged)
116  {
117  tensorHandle->SetMemoryGroup(m_MemoryManager->GetInterLayerMemoryGroup());
118  }
119  return tensorHandle;
120 }

◆ CreateTensorHandle() [2/2]

std::unique_ptr< ITensorHandle > CreateTensorHandle ( const TensorInfo &  tensorInfo,
DataLayout  dataLayout,
const bool  IsMemoryManaged = true 
) const
overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 122 of file NeonWorkloadFactory.cpp.

125 {
126  auto tensorHandle = std::make_unique<NeonTensorHandle>(tensorInfo, dataLayout);
127  if (IsMemoryManaged)
128  {
129  tensorHandle->SetMemoryGroup(m_MemoryManager->GetInterLayerMemoryGroup());
130  }
131  return tensorHandle;
132 }

◆ CreateWorkload()

std::unique_ptr< IWorkload > CreateWorkload ( LayerType  type,
const QueueDescriptor &  descriptor,
const WorkloadInfo &  info 
) const
overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 134 of file NeonWorkloadFactory.cpp.

References armnn::Abs, armnn::Activation, armnn::Addition, armnn::ArgMinMax, armnn::BatchMatMul, armnn::BatchNormalization, armnn::BatchToSpaceNd, armnn::Cast, armnn::ChannelShuffle, armnn::Comparison, armnn::Concat, armnn::Constant, armnn::ConvertBf16ToFp32, armnn::ConvertFp16ToFp32, armnn::ConvertFp32ToBf16, armnn::ConvertFp32ToFp16, armnn::Convolution2d, armnn::Convolution3d, WorkloadFactoryBase::CreateActivation(), WorkloadFactoryBase::CreateAddition(), WorkloadFactoryBase::CreateArgMinMax(), WorkloadFactoryBase::CreateBatchNormalization(), WorkloadFactoryBase::CreateBatchToSpaceNd(), WorkloadFactoryBase::CreateComparison(), WorkloadFactoryBase::CreateConcat(), WorkloadFactoryBase::CreateConstant(), WorkloadFactoryBase::CreateConvertFp16ToFp32(), WorkloadFactoryBase::CreateConvertFp32ToFp16(), WorkloadFactoryBase::CreateConvolution2d(), WorkloadFactoryBase::CreateDebug(), WorkloadFactoryBase::CreateDepthToSpace(), WorkloadFactoryBase::CreateDepthwiseConvolution2d(), WorkloadFactoryBase::CreateDequantize(), WorkloadFactoryBase::CreateDetectionPostProcess(), WorkloadFactoryBase::CreateDivision(), WorkloadFactoryBase::CreateElementwiseUnary(), WorkloadFactoryBase::CreateFloor(), WorkloadFactoryBase::CreateFullyConnected(), WorkloadFactoryBase::CreateGather(), WorkloadFactoryBase::CreateInput(), WorkloadFactoryBase::CreateInstanceNormalization(), WorkloadFactoryBase::CreateL2Normalization(), WorkloadFactoryBase::CreateLogSoftmax(), WorkloadFactoryBase::CreateLstm(), WorkloadFactoryBase::CreateMaximum(), WorkloadFactoryBase::CreateMean(), WorkloadFactoryBase::CreateMemCopy(), WorkloadFactoryBase::CreateMemImport(), WorkloadFactoryBase::CreateMinimum(), WorkloadFactoryBase::CreateMultiplication(), WorkloadFactoryBase::CreateNormalization(), WorkloadFactoryBase::CreateOutput(), WorkloadFactoryBase::CreatePad(), WorkloadFactoryBase::CreatePermute(), WorkloadFactoryBase::CreatePooling2d(), WorkloadFactoryBase::CreatePreCompiled(), WorkloadFactoryBase::CreatePrelu(), WorkloadFactoryBase::CreateQuantize(), WorkloadFactoryBase::CreateQuantizedLstm(), WorkloadFactoryBase::CreateRank(), WorkloadFactoryBase::CreateReshape(), WorkloadFactoryBase::CreateResize(), WorkloadFactoryBase::CreateSlice(), WorkloadFactoryBase::CreateSoftmax(), WorkloadFactoryBase::CreateSpaceToBatchNd(), WorkloadFactoryBase::CreateSpaceToDepth(), WorkloadFactoryBase::CreateSplitter(), WorkloadFactoryBase::CreateStack(), WorkloadFactoryBase::CreateStridedSlice(), WorkloadFactoryBase::CreateSubtraction(), WorkloadFactoryBase::CreateTranspose(), WorkloadFactoryBase::CreateTransposeConvolution2d(), armnn::Debug, armnn::DepthToSpace, armnn::DepthwiseConvolution2d, armnn::Dequantize, armnn::DetectionPostProcess, armnn::Division, armnn::ElementwiseUnary, armnn::Exp, armnn::Fill, armnn::Float32, armnn::Floor, armnn::FullyConnected, armnn::Gather, armnn::GatherNd, armnn::info, armnn::Input, armnn::InstanceNormalization, NeonBackendModelContext::IsFastMathEnabled(), armnn::L2Normalization, armnn::Log, armnn::LogicalAnd, armnn::LogicalBinary, armnn::LogicalNot, armnn::LogicalOr, armnn::LogSoftmax, armnn::Lstm, QueueDescriptor::m_Inputs, WorkloadInfo::m_InputTensorInfos, ElementwiseUnaryDescriptor::m_Operation, LogicalBinaryDescriptor::m_Operation, QueueDescriptor::m_Outputs, WorkloadInfo::m_OutputTensorInfos, QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, armnn::Maximum, armnn::Mean, armnn::MemCopy, armnn::MemImport, armnn::Minimum, armnn::Multiplication, armnn::Neg, armnn::Normalization, armnn::Output, armnn::Pad, armnn::Permute, armnn::Pooling2d, armnn::Pooling3d, armnn::PreCompiled, armnn::Prelu, armnn::QLstm, armnn::Quantize, armnn::QuantizedLstm, armnn::Rank, armnn::Reduce, armnn::Reshape, armnn::Resize, armnn::Rsqrt, armnn::Sin, armnn::Slice, armnn::Softmax, armnn::SpaceToBatchNd, armnn::SpaceToDepth, armnn::Splitter, armnn::Sqrt, armnn::Stack, armnn::StridedSlice, armnn::Subtraction, armnn::Transpose, armnn::TransposeConvolution2d, and armnn::UnidirectionalSequenceLstm.

Referenced by NeonWorkloadFactory::SupportsSubTensors().

137 {
138  switch(type)
139  {
140  case LayerType::Activation :
141  {
142  auto activationQueueDescriptor = PolymorphicDowncast<const ActivationQueueDescriptor*>(&descriptor);
143  return std::make_unique<NeonActivationWorkload>(*activationQueueDescriptor, info);
144  }
145  case LayerType::Addition :
146  {
147  auto additionQueueDescriptor = PolymorphicDowncast<const AdditionQueueDescriptor*>(&descriptor);
148  return std::make_unique<NeonAdditionWorkload>(*additionQueueDescriptor, info);
149  }
150  case LayerType::ArgMinMax :
151  {
152  auto argMinMaxQueueDescriptor = PolymorphicDowncast<const ArgMinMaxQueueDescriptor*>(&descriptor);
153  return std::make_unique<NeonArgMinMaxWorkload>(*argMinMaxQueueDescriptor, info);
154  }
155  case LayerType::BatchMatMul :
156  {
157  auto batchMatMulQueueDescriptor = PolymorphicDowncast<const BatchMatMulQueueDescriptor*>(&descriptor);
158  return std::make_unique<NeonBatchMatMulWorkload>(*batchMatMulQueueDescriptor, info);
159  }
160  case LayerType::BatchNormalization :
161  {
162  auto batchNormalizationQueueDescriptor
163  = PolymorphicDowncast<const BatchNormalizationQueueDescriptor*>(&descriptor);
164  return std::make_unique<NeonBatchNormalizationWorkload>(*batchNormalizationQueueDescriptor, info);
165  }
166  case LayerType::BatchToSpaceNd :
167  {
168  auto batchToSpaceNdQueueDescriptor
169  = PolymorphicDowncast<const BatchToSpaceNdQueueDescriptor*>(&descriptor);
170  return std::make_unique<NeonBatchToSpaceNdWorkload>(*batchToSpaceNdQueueDescriptor, info);
171  }
172  case LayerType::Cast :
173  {
174  auto castQueueDescriptor = PolymorphicDowncast<const CastQueueDescriptor*>(&descriptor);
175  return std::make_unique<NeonCastWorkload>(*castQueueDescriptor, info);
176  }
177  case LayerType::ChannelShuffle :
178  {
179  auto channelShuffleQueueDescriptor = PolymorphicDowncast<const ChannelShuffleQueueDescriptor*>(&descriptor);
180  return std::make_unique<NeonChannelShuffleWorkload>(*channelShuffleQueueDescriptor, info);
181  }
182  case LayerType::Comparison :
183  {
184  auto comparisonQueueDescriptor = PolymorphicDowncast<const ComparisonQueueDescriptor*>(&descriptor);
185  return std::make_unique<NeonComparisonWorkload>(*comparisonQueueDescriptor, info);
186  }
187  case LayerType::Concat :
188  {
189  auto concatQueueDescriptor = PolymorphicDowncast<const ConcatQueueDescriptor*>(&descriptor);
190  return std::make_unique<NeonConcatWorkload>(*concatQueueDescriptor, info);
191  }
192  case LayerType::Constant :
193  {
194  auto constantQueueDescriptor = PolymorphicDowncast<const ConstantQueueDescriptor*>(&descriptor);
195  return std::make_unique<NeonConstantWorkload>(*constantQueueDescriptor, info);
196  }
197  case LayerType::ConvertBf16ToFp32 :
198  {
199  auto convertBf16ToFp32QueueDescriptor
200  = PolymorphicDowncast<const ConvertBf16ToFp32QueueDescriptor*>(&descriptor);
201  return std::make_unique<NeonConvertBf16ToFp32Workload>(*convertBf16ToFp32QueueDescriptor, info);
202  }
203  case LayerType::ConvertFp16ToFp32 :
204  {
205  auto convertFp16ToFp32QueueDescriptor
206  = PolymorphicDowncast<const ConvertFp16ToFp32QueueDescriptor*>(&descriptor);
207  return std::make_unique<NeonConvertFp16ToFp32Workload>(*convertFp16ToFp32QueueDescriptor, info);
208  }
209  case LayerType::ConvertFp32ToBf16 :
210  {
211  auto convertFp32ToBf16QueueDescriptor
212  = PolymorphicDowncast<const ConvertFp32ToBf16QueueDescriptor*>(&descriptor);
213  return std::make_unique<NeonConvertFp32ToBf16Workload>(*convertFp32ToBf16QueueDescriptor, info);
214  }
215  case LayerType::ConvertFp32ToFp16 :
216  {
217  auto convertFp32ToFp16QueueDescriptor
218  = PolymorphicDowncast<const ConvertFp32ToFp16QueueDescriptor*>(&descriptor);
219  return std::make_unique<NeonConvertFp32ToFp16Workload>(*convertFp32ToFp16QueueDescriptor, info);
220  }
221  case LayerType::Convolution2d :
222  {
223  auto convolution2dQueueDescriptor = PolymorphicDowncast<const Convolution2dQueueDescriptor*>(&descriptor);
224 
225  bool isFastMathEnabled = false;
226  if (m_ModelContextPtr)
227  {
228  if (m_ModelContextPtr.get() != nullptr)
229  {
230  auto modelOptions = dynamic_cast<NeonBackendModelContext*>(m_ModelContextPtr.get());
231  if (modelOptions)
232  {
233  isFastMathEnabled = modelOptions->IsFastMathEnabled();
234  }
235  }
236  }
237  return std::make_unique<NeonConvolution2dWorkload>(*convolution2dQueueDescriptor,
238  info,
239  m_MemoryManager->GetIntraLayerManager(),
240  isFastMathEnabled);
241  }
242  case LayerType::Convolution3d :
243  {
244  auto convolution3dQueueDescriptor = PolymorphicDowncast<const Convolution3dQueueDescriptor*>(&descriptor);
245 
246  bool isFastMathEnabled = false;
247  if (m_ModelContextPtr)
248  {
249  if (m_ModelContextPtr.get() != nullptr)
250  {
251  auto modelOptions = dynamic_cast<NeonBackendModelContext*>(m_ModelContextPtr.get());
252  if (modelOptions)
253  {
254  isFastMathEnabled = modelOptions->IsFastMathEnabled();
255  }
256  }
257  }
258  return std::make_unique<NeonConvolution3dWorkload>(*convolution3dQueueDescriptor,
259  info,
260  m_MemoryManager->GetIntraLayerManager(),
261  isFastMathEnabled);
262  }
263  case LayerType::Debug :
264  {
265  auto debugQueueDescriptor = PolymorphicDowncast<const DebugQueueDescriptor*>(&descriptor);
266  return MakeWorkloadHelper<NullWorkload, NullWorkload>(*debugQueueDescriptor, info);
267  }
268  case LayerType::DepthToSpace :
269  {
270  auto depthToSpaceQueueDescriptor = PolymorphicDowncast<const DepthToSpaceQueueDescriptor*>(&descriptor);
271  return std::make_unique<NeonDepthToSpaceWorkload>(*depthToSpaceQueueDescriptor, info);
272  }
273  case LayerType::DepthwiseConvolution2d :
274  {
275  auto depthwiseConvolution2dQueueDescriptor
276  = PolymorphicDowncast<const DepthwiseConvolution2dQueueDescriptor*>(&descriptor);
277  return std::make_unique<NeonDepthwiseConvolutionWorkload>(*depthwiseConvolution2dQueueDescriptor, info);
278  }
279  case LayerType::Dequantize :
280  {
281  auto dequantizeQueueDescriptor = PolymorphicDowncast<const DequantizeQueueDescriptor*>(&descriptor);
282  return std::make_unique<NeonDequantizeWorkload>(*dequantizeQueueDescriptor, info);
283  }
284  case LayerType::DetectionPostProcess :
285  {
286  auto detectionPostProcessQueueDescriptor
287  = PolymorphicDowncast<const DetectionPostProcessQueueDescriptor*>(&descriptor);
288  return MakeWorkloadHelper<NullWorkload, NullWorkload>(*detectionPostProcessQueueDescriptor, info);
289  }
290  case LayerType::Division :
291  {
292  auto divisionQueueDescriptor = PolymorphicDowncast<const DivisionQueueDescriptor*>(&descriptor);
293  return std::make_unique<NeonDivisionWorkload>(*divisionQueueDescriptor, info);
294  }
295  case LayerType::ElementwiseUnary :
296  {
297  auto elementwiseUnaryQueueDescriptor
298  = PolymorphicDowncast<const ElementwiseUnaryQueueDescriptor*>(&descriptor);
299 
300  switch(elementwiseUnaryQueueDescriptor->m_Parameters.m_Operation)
301  {
302  case UnaryOperation::Abs:
303  {
304  AbsQueueDescriptor absQueueDescriptor;
305  absQueueDescriptor.m_Inputs = elementwiseUnaryQueueDescriptor->m_Inputs;
306  absQueueDescriptor.m_Outputs = elementwiseUnaryQueueDescriptor->m_Outputs;
307 
308  return std::make_unique<NeonAbsWorkload>(absQueueDescriptor, info);
309  }
310  case UnaryOperation::Exp:
311  return std::make_unique<NeonExpWorkload>(*elementwiseUnaryQueueDescriptor, info);
312  case UnaryOperation::LogicalNot:
313  return std::make_unique<NeonLogicalNotWorkload>(*elementwiseUnaryQueueDescriptor, info);
314  case UnaryOperation::Log:
315  return std::make_unique<NeonLogWorkload>(*elementwiseUnaryQueueDescriptor, info);
316  case UnaryOperation::Neg:
317  return std::make_unique<NeonNegWorkload>(*elementwiseUnaryQueueDescriptor, info);
318  case UnaryOperation::Rsqrt:
319  {
320  RsqrtQueueDescriptor rsqrtQueueDescriptor;
321  rsqrtQueueDescriptor.m_Inputs = elementwiseUnaryQueueDescriptor->m_Inputs;
322  rsqrtQueueDescriptor.m_Outputs = elementwiseUnaryQueueDescriptor->m_Outputs;
323 
324  return std::make_unique<NeonRsqrtWorkload>(rsqrtQueueDescriptor, info);
325  }
326  case UnaryOperation::Sin:
327  return std::make_unique<NeonSinWorkload>(*elementwiseUnaryQueueDescriptor, info);
328  case UnaryOperation::Sqrt:
329  return std::make_unique<NeonSqrtWorkload>(*elementwiseUnaryQueueDescriptor, info);
330  default:
331  return nullptr;
332  }
333  }
334  case LayerType::Fill :
335  {
336  auto fillQueueDescriptor = PolymorphicDowncast<const FillQueueDescriptor*>(&descriptor);
337  return std::make_unique<NeonFillWorkload>(*fillQueueDescriptor, info);
338  }
339  case LayerType::Floor :
340  {
341  auto floorQueueDescriptor = PolymorphicDowncast<const FloorQueueDescriptor*>(&descriptor);
342  return MakeWorkloadHelper<NeonFloorFloatWorkload, NullWorkload>(*floorQueueDescriptor, info);
343  }
344  case LayerType::FullyConnected :
345  {
346  auto fullyConnectedQueueDescriptor = PolymorphicDowncast<const FullyConnectedQueueDescriptor*>(&descriptor);
347  return std::make_unique<NeonFullyConnectedWorkload>(*fullyConnectedQueueDescriptor,
348  info,
349  m_MemoryManager->GetIntraLayerManager());
350  }
351  case LayerType::Gather :
352  {
353  auto gatherQueueDescriptor = PolymorphicDowncast<const GatherQueueDescriptor*>(&descriptor);
354  return std::make_unique<NeonGatherWorkload>(*gatherQueueDescriptor, info);
355  }
356  case LayerType::GatherNd :
357  {
358  auto gatherNdQueueDescriptor = PolymorphicDowncast<const GatherNdQueueDescriptor*>(&descriptor);
359  return std::make_unique<NeonGatherNdWorkload>(*gatherNdQueueDescriptor, info);
360  }
361  case LayerType::Input :
362  {
363  auto inputQueueDescriptor = PolymorphicDowncast<const InputQueueDescriptor*>(&descriptor);
364  return std::make_unique<CopyMemGenericWorkload>(*inputQueueDescriptor, info);
365  }
366  case LayerType::InstanceNormalization :
367  {
368  auto instanceNormalizationQueueDescriptor
369  = PolymorphicDowncast<const InstanceNormalizationQueueDescriptor*>(&descriptor);
370  return std::make_unique<NeonInstanceNormalizationWorkload>(*instanceNormalizationQueueDescriptor, info);
371  }
372  case LayerType::L2Normalization :
373  {
374  auto l2NormalizationQueueDescriptor
375  = PolymorphicDowncast<const L2NormalizationQueueDescriptor*>(&descriptor);
376  return MakeWorkloadHelper<NeonL2NormalizationFloatWorkload, NullWorkload>
377  (*l2NormalizationQueueDescriptor, info, m_MemoryManager->GetIntraLayerManager());
378  }
379  case LayerType::LogSoftmax :
380  {
381  auto logSoftmaxQueueDescriptor = PolymorphicDowncast<const LogSoftmaxQueueDescriptor*>(&descriptor);
382  return std::make_unique<NeonLogSoftmaxWorkload>(*logSoftmaxQueueDescriptor,
383  info,
384  m_MemoryManager->GetIntraLayerManager());
385  }
386  case LayerType::LogicalBinary :
387  {
388  auto logicalBinaryQueueDescriptor = PolymorphicDowncast<const LogicalBinaryQueueDescriptor*>(&descriptor);
389 
390  switch(logicalBinaryQueueDescriptor->m_Parameters.m_Operation)
391  {
392  case LogicalBinaryOperation::LogicalAnd:
393  return std::make_unique<NeonLogicalAndWorkload>(*logicalBinaryQueueDescriptor, info);
394  case LogicalBinaryOperation::LogicalOr:
395  return std::make_unique<NeonLogicalOrWorkload>(*logicalBinaryQueueDescriptor, info);
396  default:
397  return nullptr;
398  }
399  }
400  case LayerType::Lstm :
401  {
402  auto lstmQueueDescriptor = PolymorphicDowncast<const LstmQueueDescriptor*>(&descriptor);
403  return MakeWorkloadHelper<NeonLstmFloatWorkload, NullWorkload>(*lstmQueueDescriptor, info);
404  }
405  case LayerType::Maximum :
406  {
407  auto maximumQueueDescriptor = PolymorphicDowncast<const MaximumQueueDescriptor*>(&descriptor);
408  return std::make_unique<NeonMaximumWorkload>(*maximumQueueDescriptor, info);
409  }
410  case LayerType::Mean :
411  {
412  auto meanQueueDescriptor = PolymorphicDowncast<const MeanQueueDescriptor*>(&descriptor);
413  return std::make_unique<NeonMeanWorkload>(*meanQueueDescriptor, info);
414  }
415  case LayerType::MemCopy :
416  {
417  auto memCopyQueueDescriptor = PolymorphicDowncast<const MemCopyQueueDescriptor*>(&descriptor);
418  if (memCopyQueueDescriptor->m_Inputs.empty() || !memCopyQueueDescriptor->m_Inputs[0])
419  {
420  throw InvalidArgumentException("NeonWorkloadFactory: Invalid null input for MemCopy workload");
421  }
422  return MakeWorkloadHelper<CopyMemGenericWorkload, CopyMemGenericWorkload>(*memCopyQueueDescriptor, info);
423  }
424  case LayerType::MemImport :
425  {
426  auto memImportQueueDescriptor = PolymorphicDowncast<const MemImportQueueDescriptor*>(&descriptor);
427  if (memImportQueueDescriptor->m_Inputs.empty() || !memImportQueueDescriptor->m_Inputs[0])
428  {
429  throw InvalidArgumentException("NeonWorkloadFactory: Invalid null input for MemImport workload");
430  }
431  return std::make_unique<ImportMemGenericWorkload>(*memImportQueueDescriptor, info);
432  }
433  case LayerType::Minimum :
434  {
435  auto minimumQueueDescriptor = PolymorphicDowncast<const MinimumQueueDescriptor*>(&descriptor);
436  return std::make_unique<NeonMinimumWorkload>(*minimumQueueDescriptor, info);
437  }
438  case LayerType::Multiplication :
439  {
440  auto multiplicationQueueDescriptor = PolymorphicDowncast<const MultiplicationQueueDescriptor*>(&descriptor);
441  return std::make_unique<NeonMultiplicationWorkload>(*multiplicationQueueDescriptor, info);
442  }
443  case LayerType::Normalization :
444  {
445  auto normalizationQueueDescriptor = PolymorphicDowncast<const NormalizationQueueDescriptor*>(&descriptor);
446  return MakeWorkloadHelper<NeonNormalizationFloatWorkload, NullWorkload>
447  (*normalizationQueueDescriptor, info, m_MemoryManager->GetIntraLayerManager());
448  }
449  case LayerType::Output :
450  {
451  auto outputQueueDescriptor = PolymorphicDowncast<const OutputQueueDescriptor*>(&descriptor);
452  return std::make_unique<CopyMemGenericWorkload>(*outputQueueDescriptor, info);
453  }
454  case LayerType::Pad :
455  {
456  auto padQueueDescriptor = PolymorphicDowncast<const PadQueueDescriptor*>(&descriptor);
457  return std::make_unique<NeonPadWorkload>(*padQueueDescriptor, info);
458  }
459  case LayerType::Permute :
460  {
461  auto permuteQueueDescriptor = PolymorphicDowncast<const PermuteQueueDescriptor*>(&descriptor);
462  return std::make_unique<NeonPermuteWorkload>(*permuteQueueDescriptor, info);
463  }
464  case LayerType::Pooling2d :
465  {
466  auto pooling2dQueueDescriptor = PolymorphicDowncast<const Pooling2dQueueDescriptor*>(&descriptor);
467  return std::make_unique<NeonPooling2dWorkload>(*pooling2dQueueDescriptor, info);
468  }
469  case LayerType::Pooling3d :
470  {
471  auto pooling3dQueueDescriptor = PolymorphicDowncast<const Pooling3dQueueDescriptor*>(&descriptor);
472  return std::make_unique<NeonPooling3dWorkload>(*pooling3dQueueDescriptor, info);
473  }
474  case LayerType::PreCompiled :
475  {
476  auto preCompiledQueueDescriptor = PolymorphicDowncast<const PreCompiledQueueDescriptor*>(&descriptor);
477  return MakeWorkloadHelper<NullWorkload, NullWorkload>(*preCompiledQueueDescriptor, info);
478  }
479  case LayerType::Prelu :
480  {
481  auto preluQueueDescriptor = PolymorphicDowncast<const PreluQueueDescriptor*>(&descriptor);
482  return std::make_unique<NeonPreluWorkload>(*preluQueueDescriptor, info);
483  }
484  case LayerType::QLstm :
485  {
486  auto qLstmQueueDescriptor = PolymorphicDowncast<const QLstmQueueDescriptor*>(&descriptor);
487  return std::make_unique<NeonQLstmWorkload>(*qLstmQueueDescriptor, info);
488  }
489  case LayerType::Quantize :
490  {
491  auto quantizeQueueDescriptor = PolymorphicDowncast<const QuantizeQueueDescriptor*>(&descriptor);
492  return std::make_unique<NeonQuantizeWorkload>(*quantizeQueueDescriptor, info);
493  }
494  case LayerType::QuantizedLstm :
495  {
496  auto quantizedLstmQueueDescriptor = PolymorphicDowncast<const QuantizedLstmQueueDescriptor*>(&descriptor);
497  return std::make_unique<NeonQuantizedLstmWorkload>(*quantizedLstmQueueDescriptor, info);
498  }
499  case LayerType::Rank :
500  {
501  auto rankQueueDescriptor = PolymorphicDowncast<const RankQueueDescriptor*>(&descriptor);
502  return std::make_unique<NeonRankWorkload>(*rankQueueDescriptor, info);
503  }
504  case LayerType::Reduce :
505  {
506  auto reduceQueueDescriptor = PolymorphicDowncast<const ReduceQueueDescriptor*>(&descriptor);
507  return std::make_unique<NeonReduceWorkload>(*reduceQueueDescriptor, info);
508  }
509  case LayerType::Reshape :
510  {
511  auto reshapeQueueDescriptor = PolymorphicDowncast<const ReshapeQueueDescriptor*>(&descriptor);
512  return std::make_unique<NeonReshapeWorkload>(*reshapeQueueDescriptor, info);
513  }
514  case LayerType::Resize :
515  {
516  auto resizeQueueDescriptor = PolymorphicDowncast<const ResizeQueueDescriptor*>(&descriptor);
517  return std::make_unique<NeonResizeWorkload>(*resizeQueueDescriptor, info);
518  }
519  case LayerType::Slice :
520  {
521  auto sliceQueueDescriptor = PolymorphicDowncast<const SliceQueueDescriptor*>(&descriptor);
522  return std::make_unique<NeonSliceWorkload>(*sliceQueueDescriptor, info);
523  }
524  case LayerType::Softmax :
525  {
526  auto softmaxQueueDescriptor = PolymorphicDowncast<const SoftmaxQueueDescriptor*>(&descriptor);
527  return std::make_unique<NeonSoftmaxWorkload>(*softmaxQueueDescriptor,
528  info,
529  m_MemoryManager->GetIntraLayerManager());
530  }
531  case LayerType::SpaceToBatchNd :
532  {
533  auto spaceToBatchNdQueueDescriptor
534  = PolymorphicDowncast<const SpaceToBatchNdQueueDescriptor*>(&descriptor);
535  return std::make_unique<NeonSpaceToBatchNdWorkload>(*spaceToBatchNdQueueDescriptor, info);
536  }
537  case LayerType::SpaceToDepth :
538  {
539  auto spaceToDepthQueueDescriptor = PolymorphicDowncast<const SpaceToDepthQueueDescriptor*>(&descriptor);
540  return std::make_unique<NeonSpaceToDepthWorkload>(*spaceToDepthQueueDescriptor, info);
541  }
542  case LayerType::Splitter :
543  {
544  auto splitterQueueDescriptor = PolymorphicDowncast<const SplitterQueueDescriptor*>(&descriptor);
545  return std::make_unique<NeonSplitterWorkload>(*splitterQueueDescriptor, info);
546  }
547  case LayerType::Stack :
548  {
549  auto stackQueueDescriptor = PolymorphicDowncast<const StackQueueDescriptor*>(&descriptor);
550  return std::make_unique<NeonStackWorkload>(*stackQueueDescriptor, info);
551  }
552  case LayerType::StridedSlice :
553  {
554  auto stridedSliceQueueDescriptor = PolymorphicDowncast<const StridedSliceQueueDescriptor*>(&descriptor);
555  return std::make_unique<NeonStridedSliceWorkload>(*stridedSliceQueueDescriptor, info);
556  }
557  case LayerType::Subtraction :
558  {
559  auto subtractionQueueDescriptor = PolymorphicDowncast<const SubtractionQueueDescriptor*>(&descriptor);
560  return std::make_unique<NeonSubtractionWorkload>(*subtractionQueueDescriptor, info);
561  }
562  case LayerType::Transpose :
563  {
564  auto transposeQueueDescriptor = PolymorphicDowncast<const TransposeQueueDescriptor*>(&descriptor);
565  return std::make_unique<NeonTransposeWorkload>(*transposeQueueDescriptor, info);
566  }
567  case LayerType::TransposeConvolution2d :
568  {
569  auto transposeConvolution2dQueueDescriptor
570  = PolymorphicDowncast<const TransposeConvolution2dQueueDescriptor*>(&descriptor);
571  return std::make_unique<NeonTransposeConvolution2dWorkload>(*transposeConvolution2dQueueDescriptor,
572  info,
573  m_MemoryManager->GetIntraLayerManager());
574  }
575  case LayerType::UnidirectionalSequenceLstm :
576  {
577  auto desc = PolymorphicDowncast<const UnidirectionalSequenceLstmQueueDescriptor*>(&descriptor);
578 
579  if ((info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Float32) &&
580  (info.m_InputTensorInfos[1].GetDataType() == armnn::DataType::Float32) &&
581  (info.m_InputTensorInfos[2].GetDataType() == armnn::DataType::Float32) &&
582  (info.m_OutputTensorInfos[0].GetDataType() == armnn::DataType::Float32) &&
583  (info.m_OutputTensorInfos[1].GetDataType() == armnn::DataType::Float32) &&
584  (info.m_OutputTensorInfos[2].GetDataType() == armnn::DataType::Float32))
585  {
586  return std::make_unique<NeonUnidirectionalSequenceLstmFloatWorkload>(*desc, info);
587  }
588  else
589  {
590  return std::make_unique<NeonUnidirectionalSequenceLstmWorkload>(*desc, info);
591  }
592  }
593  default:
594  return nullptr;
595  }
596 }

◆ GetBackendId()

const BackendId & GetBackendId ( ) const
overridevirtual

Implements IWorkloadFactory.

Definition at line 49 of file NeonWorkloadFactory.cpp.

References NeonBackendModelContext::GetNumberOfThreads().

50 {
51  return s_Id;
52 }

◆ IsLayerSupported() [1/2]

bool IsLayerSupported ( const Layer &  layer,
Optional< DataType >  dataType,
std::string &  outReasonIfUnsupported 
)
static

Definition at line 34 of file NeonWorkloadFactory.cpp.

References IWorkloadFactory::IsLayerSupported().

37 {
38  return IWorkloadFactory::IsLayerSupported(s_Id, layer, dataType, outReasonIfUnsupported);
39 }
armnn::IWorkloadFactory::IsLayerSupported
static bool IsLayerSupported(const BackendId &backendId, const IConnectableLayer &layer, Optional< DataType > dataType, std::string &outReasonIfUnsupported)
Definition: WorkloadFactory.cpp:1532

◆ IsLayerSupported() [2/2]

bool IsLayerSupported ( const IConnectableLayer &  layer,
Optional< DataType >  dataType,
std::string &  outReasonIfUnsupported,
const ModelOptions &  modelOptions 
)
static

Definition at line 41 of file NeonWorkloadFactory.cpp.

References IWorkloadFactory::IsLayerSupported().

45 {
46  return IWorkloadFactory::IsLayerSupported(s_Id, layer, dataType, outReasonIfUnsupported, modelOptions);
47 }
armnn::IWorkloadFactory::IsLayerSupported
static bool IsLayerSupported(const BackendId &backendId, const IConnectableLayer &layer, Optional< DataType > dataType, std::string &outReasonIfUnsupported)
Definition: WorkloadFactory.cpp:1532

◆ SupportsSubTensors()

bool SupportsSubTensors ( ) const
inlineoverridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 39 of file NeonWorkloadFactory.hpp.

References ARMNN_DEPRECATED_MSG, ARMNN_DEPRECATED_MSG_REMOVAL_DATE, WorkloadFactoryBase::CreateActivation(), WorkloadFactoryBase::CreateAddition(), WorkloadFactoryBase::CreateArgMinMax(), WorkloadFactoryBase::CreateBatchNormalization(), WorkloadFactoryBase::CreateBatchToSpaceNd(), WorkloadFactoryBase::CreateComparison(), WorkloadFactoryBase::CreateConcat(), WorkloadFactoryBase::CreateConstant(), WorkloadFactoryBase::CreateConvertFp16ToFp32(), WorkloadFactoryBase::CreateConvertFp32ToFp16(), WorkloadFactoryBase::CreateConvolution2d(), WorkloadFactoryBase::CreateDebug(), WorkloadFactoryBase::CreateDepthToSpace(), WorkloadFactoryBase::CreateDepthwiseConvolution2d(), WorkloadFactoryBase::CreateDequantize(), WorkloadFactoryBase::CreateDetectionPostProcess(), WorkloadFactoryBase::CreateDivision(), WorkloadFactoryBase::CreateElementwiseUnary(), WorkloadFactoryBase::CreateFloor(), WorkloadFactoryBase::CreateFullyConnected(), WorkloadFactoryBase::CreateGather(), WorkloadFactoryBase::CreateInput(), WorkloadFactoryBase::CreateInstanceNormalization(), WorkloadFactoryBase::CreateL2Normalization(), WorkloadFactoryBase::CreateLogSoftmax(), WorkloadFactoryBase::CreateLstm(), WorkloadFactoryBase::CreateMaximum(), WorkloadFactoryBase::CreateMean(), WorkloadFactoryBase::CreateMemCopy(), WorkloadFactoryBase::CreateMemImport(), WorkloadFactoryBase::CreateMinimum(), WorkloadFactoryBase::CreateMultiplication(), WorkloadFactoryBase::CreateNormalization(), WorkloadFactoryBase::CreateOutput(), WorkloadFactoryBase::CreatePad(), WorkloadFactoryBase::CreatePermute(), WorkloadFactoryBase::CreatePooling2d(), WorkloadFactoryBase::CreatePreCompiled(), WorkloadFactoryBase::CreatePrelu(), WorkloadFactoryBase::CreateQuantize(), WorkloadFactoryBase::CreateQuantizedLstm(), WorkloadFactoryBase::CreateRank(), WorkloadFactoryBase::CreateReshape(), WorkloadFactoryBase::CreateResize(), WorkloadFactoryBase::CreateSlice(), WorkloadFactoryBase::CreateSoftmax(), WorkloadFactoryBase::CreateSpaceToBatchNd(), WorkloadFactoryBase::CreateSpaceToDepth(), WorkloadFactoryBase::CreateSplitter(), WorkloadFactoryBase::CreateStack(), WorkloadFactoryBase::CreateStridedSlice(), NeonWorkloadFactory::CreateSubTensorHandle(), WorkloadFactoryBase::CreateSubtraction(), NeonWorkloadFactory::CreateTensorHandle(), WorkloadFactoryBase::CreateTranspose(), WorkloadFactoryBase::CreateTransposeConvolution2d(), NeonWorkloadFactory::CreateWorkload(), armnn::Info, and armnn::info.

39 { return true; }
The documentation for this class was generated from the following files: