21.11
|
#include <SampleDynamicWorkloadFactory.hpp>
Public Member Functions | |
SampleDynamicWorkloadFactory (const std::shared_ptr< SampleMemoryManager > &memoryManager) | |
SampleDynamicWorkloadFactory () | |
~SampleDynamicWorkloadFactory () | |
const armnn::BackendId & | GetBackendId () const override |
bool | SupportsSubTensors () const override |
std::unique_ptr< armnn::ITensorHandle > | CreateSubTensorHandle (armnn::ITensorHandle &parent, armnn::TensorShape const &subTensorShape, unsigned int const *subTensorOrigin) const override |
std::unique_ptr< armnn::ITensorHandle > | CreateTensorHandle (const armnn::TensorInfo &tensorInfo, const bool IsMemoryManaged=true) const override |
std::unique_ptr< armnn::ITensorHandle > | CreateTensorHandle (const armnn::TensorInfo &tensorInfo, armnn::DataLayout dataLayout, const bool IsMemoryManaged=true) const override |
std::unique_ptr< armnn::IWorkload > | CreateAddition (const armnn::AdditionQueueDescriptor &descriptor, const armnn::WorkloadInfo &info) const override |
std::unique_ptr< armnn::IWorkload > | CreateInput (const armnn::InputQueueDescriptor &descriptor, const armnn::WorkloadInfo &info) const override |
std::unique_ptr< armnn::IWorkload > | CreateOutput (const armnn::OutputQueueDescriptor &descriptor, const armnn::WorkloadInfo &info) const override |
![]() | |
virtual | ~IWorkloadFactory () |
virtual void | AfterWorkloadsCreated () |
virtual std::unique_ptr< IWorkload > | CreateActivation (const ActivationQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateArgMinMax (const ArgMinMaxQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateBatchNormalization (const BatchNormalizationQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateBatchToSpaceNd (const BatchToSpaceNdQueueDescriptor &descriptor, const WorkloadInfo &Info) const |
virtual std::unique_ptr< IWorkload > | CreateCast (const CastQueueDescriptor &descriptor, const WorkloadInfo &Info) const |
virtual std::unique_ptr< IWorkload > | CreateChannelShuffle (const ChannelShuffleQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateComparison (const ComparisonQueueDescriptor &descriptor, const WorkloadInfo &Info) const |
virtual std::unique_ptr< IWorkload > | CreateConcat (const ConcatQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateConstant (const ConstantQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateConvertBf16ToFp32 (const ConvertBf16ToFp32QueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateConvertFp16ToFp32 (const ConvertFp16ToFp32QueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateConvertFp32ToBf16 (const ConvertFp32ToBf16QueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateConvertFp32ToFp16 (const ConvertFp32ToFp16QueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateConvolution2d (const Convolution2dQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateConvolution3d (const Convolution3dQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateDebug (const DebugQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateDepthToSpace (const DepthToSpaceQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateDepthwiseConvolution2d (const DepthwiseConvolution2dQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateDequantize (const DequantizeQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateDetectionPostProcess (const DetectionPostProcessQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateDivision (const DivisionQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateElementwiseUnary (const ElementwiseUnaryQueueDescriptor &descriptor, const WorkloadInfo &Info) const |
virtual std::unique_ptr< IWorkload > | CreateFakeQuantization (const FakeQuantizationQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateFill (const FillQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateFloor (const FloorQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateFullyConnected (const FullyConnectedQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateGather (const GatherQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateInstanceNormalization (const InstanceNormalizationQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateL2Normalization (const L2NormalizationQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateLogicalBinary (const LogicalBinaryQueueDescriptor &descriptor, const WorkloadInfo &Info) const |
virtual std::unique_ptr< IWorkload > | CreateLogicalUnary (const ElementwiseUnaryQueueDescriptor &descriptor, const WorkloadInfo &Info) const |
virtual std::unique_ptr< IWorkload > | CreateLogSoftmax (const LogSoftmaxQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateLstm (const LstmQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateMaximum (const MaximumQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateMean (const MeanQueueDescriptor &descriptor, const WorkloadInfo &Info) const |
virtual std::unique_ptr< IWorkload > | CreateMemCopy (const MemCopyQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateMemImport (const MemImportQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateMerge (const MergeQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateMinimum (const MinimumQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateMultiplication (const MultiplicationQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateNormalization (const NormalizationQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreatePad (const PadQueueDescriptor &descriptor, const WorkloadInfo &Info) const |
virtual std::unique_ptr< IWorkload > | CreatePermute (const PermuteQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreatePooling2d (const Pooling2dQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreatePreCompiled (const PreCompiledQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreatePrelu (const PreluQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateQuantize (const QuantizeQueueDescriptor &descriptor, const WorkloadInfo &Info) const |
virtual std::unique_ptr< IWorkload > | CreateQLstm (const QLstmQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateQuantizedLstm (const QuantizedLstmQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateRank (const RankQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateReduce (const ReduceQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateReshape (const ReshapeQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateResize (const ResizeQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateShape (const ShapeQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateSlice (const SliceQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateSoftmax (const SoftmaxQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateSpaceToBatchNd (const SpaceToBatchNdQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateSpaceToDepth (const SpaceToDepthQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateSubtraction (const SubtractionQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateSplitter (const SplitterQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateStack (const StackQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateStridedSlice (const StridedSliceQueueDescriptor &descriptor, const WorkloadInfo &Info) const |
virtual std::unique_ptr< IWorkload > | CreateSwitch (const SwitchQueueDescriptor &descriptor, const WorkloadInfo &Info) const |
virtual std::unique_ptr< IWorkload > | CreateTranspose (const TransposeQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateTransposeConvolution2d (const TransposeConvolution2dQueueDescriptor &descriptor, const WorkloadInfo &info) const |
virtual std::unique_ptr< IWorkload > | CreateUnidirectionalSequenceLstm (const UnidirectionalSequenceLstmQueueDescriptor &descriptor, const WorkloadInfo &info) const |
Static Public Member Functions | |
static bool | IsLayerSupported (const armnn::IConnectableLayer &layer, armnn::Optional< armnn::DataType > dataType, std::string &outReasonIfUnsupported) |
![]() | |
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) |
Definition at line 16 of file SampleDynamicWorkloadFactory.hpp.
|
explicit |
Definition at line 22 of file SampleDynamicWorkloadFactory.cpp.
Definition at line 27 of file SampleDynamicWorkloadFactory.cpp.
|
inline |
Definition at line 22 of file SampleDynamicWorkloadFactory.hpp.
References SampleDynamicWorkloadFactory::GetBackendId(), and SampleDynamicWorkloadFactory::IsLayerSupported().
|
overridevirtual |
Reimplemented from IWorkloadFactory.
Definition at line 59 of file SampleDynamicWorkloadFactory.cpp.
Referenced by SampleDynamicWorkloadFactory::CreateSubTensorHandle().
|
overridevirtual |
Implements IWorkloadFactory.
Definition at line 66 of file SampleDynamicWorkloadFactory.cpp.
Referenced by SampleDynamicWorkloadFactory::CreateSubTensorHandle().
|
overridevirtual |
Reimplemented from IWorkloadFactory.
Definition at line 73 of file SampleDynamicWorkloadFactory.cpp.
Referenced by SampleDynamicWorkloadFactory::CreateSubTensorHandle().
|
inlineoverridevirtual |
Implements IWorkloadFactory.
Definition at line 33 of file SampleDynamicWorkloadFactory.hpp.
References ARMNN_DEPRECATED_MSG, SampleDynamicWorkloadFactory::CreateAddition(), SampleDynamicWorkloadFactory::CreateInput(), SampleDynamicWorkloadFactory::CreateOutput(), SampleDynamicWorkloadFactory::CreateTensorHandle(), and armnn::IgnoreUnused().
|
overridevirtual |
Implements IWorkloadFactory.
Definition at line 44 of file SampleDynamicWorkloadFactory.cpp.
Referenced by SampleDynamicWorkloadFactory::CreateSubTensorHandle().
|
overridevirtual |
Implements IWorkloadFactory.
Definition at line 51 of file SampleDynamicWorkloadFactory.cpp.
|
overridevirtual |
Implements IWorkloadFactory.
Definition at line 32 of file SampleDynamicWorkloadFactory.cpp.
Referenced by SampleDynamicWorkloadFactory::~SampleDynamicWorkloadFactory().
|
static |
Definition at line 37 of file SampleDynamicWorkloadFactory.cpp.
Referenced by SampleDynamicWorkloadFactory::~SampleDynamicWorkloadFactory().
|
inlineoverridevirtual |
Implements IWorkloadFactory.
Definition at line 30 of file SampleDynamicWorkloadFactory.hpp.
References ARMNN_DEPRECATED_MSG.