IWorkloadFactory Class Reference

#include <WorkloadFactory.hpp>

Inheritance diagram for IWorkloadFactory:

Public Member Functions
virtual	~IWorkloadFactory ()
virtual void	AfterWorkloadsCreated ()
virtual const BackendId &	GetBackendId () const =0
virtual bool	SupportsSubTensors () const =0
virtual std::unique_ptr< ITensorHandle >	CreateSubTensorHandle (ITensorHandle &parent, TensorShape const &subTensorShape, unsigned int const *subTensorOrigin) const =0
virtual std::unique_ptr< ITensorHandle >	CreateTensorHandle (const TensorInfo &tensorInfo, const bool IsMemoryManaged=true) const =0
virtual std::unique_ptr< ITensorHandle >	CreateTensorHandle (const TensorInfo &tensorInfo, DataLayout dataLayout, const bool IsMemoryManaged=true) const =0
virtual std::unique_ptr< IWorkload >	CreateWorkload (LayerType type, const QueueDescriptor &descriptor, const WorkloadInfo &info) const =0
	Backends should implement their own CreateWorkload function with a switch statement.

Static Public Member Functions
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 22 of file WorkloadFactory.hpp.

Constructor & Destructor Documentation

~IWorkloadFactory()

virtual ~IWorkloadFactory ( )

inlinevirtual

Definition at line 25 of file WorkloadFactory.hpp.

{ }

Member Function Documentation

AfterWorkloadsCreated()

virtual void AfterWorkloadsCreated ( )

inlinevirtual

Reimplemented in ClWorkloadFactory.

Definition at line 27 of file WorkloadFactory.hpp.

{};

CreateSubTensorHandle()

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

pure virtual

Implemented in ClWorkloadFactory, NeonWorkloadFactory, RefWorkloadFactory, TosaRefWorkloadFactory, WorkloadFactoryBase, and SampleDynamicWorkloadFactory.

References ARMNN_DEPRECATED_MSG, and CreateSubTensorHandle().

Referenced by CreateSubTensorHandle().

CreateTensorHandle() [1/2]

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

pure virtual

Implemented in ClWorkloadFactory, NeonWorkloadFactory, RefWorkloadFactory, TosaRefWorkloadFactory, WorkloadFactoryBase, and SampleDynamicWorkloadFactory.

References ARMNN_DEPRECATED_MSG, and CreateTensorHandle().

Referenced by CreateTensorHandle(), CreateTensorHandle(), and OutputHandler::CreateTensorHandles().

CreateTensorHandle() [2/2]

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

pure virtual

Implemented in ClWorkloadFactory, NeonWorkloadFactory, RefWorkloadFactory, TosaRefWorkloadFactory, WorkloadFactoryBase, and SampleDynamicWorkloadFactory.

References CreateTensorHandle().

CreateWorkload()

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

pure virtual

Backends should implement their own CreateWorkload function with a switch statement.

The case for the switch should be the LayerType and based on that they will call their specific workload creation functionality.

Implemented in ClWorkloadFactory, NeonWorkloadFactory, RefWorkloadFactory, TosaRefWorkloadFactory, WorkloadFactoryBase, and SampleDynamicWorkloadFactory.

References CreateWorkload(), and armnn::info.

Referenced by AbsLayer::CreateWorkload(), ActivationLayer::CreateWorkload(), AdditionLayer::CreateWorkload(), ArgMinMaxLayer::CreateWorkload(), BatchMatMulLayer::CreateWorkload(), BatchNormalizationLayer::CreateWorkload(), BatchToSpaceNdLayer::CreateWorkload(), BroadcastToLayer::CreateWorkload(), CastLayer::CreateWorkload(), ChannelShuffleLayer::CreateWorkload(), ComparisonLayer::CreateWorkload(), ConcatLayer::CreateWorkload(), ConstantLayer::CreateWorkload(), ConvertFp16ToFp32Layer::CreateWorkload(), ConvertFp32ToFp16Layer::CreateWorkload(), Convolution2dLayer::CreateWorkload(), Convolution3dLayer::CreateWorkload(), DebugLayer::CreateWorkload(), DepthToSpaceLayer::CreateWorkload(), DepthwiseConvolution2dLayer::CreateWorkload(), DequantizeLayer::CreateWorkload(), DetectionPostProcessLayer::CreateWorkload(), DivisionLayer::CreateWorkload(), ElementwiseBinaryLayer::CreateWorkload(), ElementwiseUnaryLayer::CreateWorkload(), FakeQuantizationLayer::CreateWorkload(), FillLayer::CreateWorkload(), FloorLayer::CreateWorkload(), FullyConnectedLayer::CreateWorkload(), FusedLayer::CreateWorkload(), GatherLayer::CreateWorkload(), GatherNdLayer::CreateWorkload(), InstanceNormalizationLayer::CreateWorkload(), CreateWorkload(), L2NormalizationLayer::CreateWorkload(), LogicalBinaryLayer::CreateWorkload(), LogSoftmaxLayer::CreateWorkload(), LstmLayer::CreateWorkload(), MaximumLayer::CreateWorkload(), MeanLayer::CreateWorkload(), MinimumLayer::CreateWorkload(), MultiplicationLayer::CreateWorkload(), NormalizationLayer::CreateWorkload(), PadLayer::CreateWorkload(), PermuteLayer::CreateWorkload(), Pooling2dLayer::CreateWorkload(), Pooling3dLayer::CreateWorkload(), PreCompiledLayer::CreateWorkload(), PreluLayer::CreateWorkload(), QLstmLayer::CreateWorkload(), QuantizedLstmLayer::CreateWorkload(), QuantizeLayer::CreateWorkload(), RankLayer::CreateWorkload(), ReduceLayer::CreateWorkload(), ReshapeLayer::CreateWorkload(), ResizeLayer::CreateWorkload(), ReverseV2Layer::CreateWorkload(), RsqrtLayer::CreateWorkload(), ScatterNdLayer::CreateWorkload(), ShapeLayer::CreateWorkload(), SliceLayer::CreateWorkload(), SoftmaxLayer::CreateWorkload(), SpaceToBatchNdLayer::CreateWorkload(), SpaceToDepthLayer::CreateWorkload(), SplitterLayer::CreateWorkload(), StackLayer::CreateWorkload(), StridedSliceLayer::CreateWorkload(), SubtractionLayer::CreateWorkload(), SwitchLayer::CreateWorkload(), TileLayer::CreateWorkload(), TransposeConvolution2dLayer::CreateWorkload(), TransposeLayer::CreateWorkload(), and UnidirectionalSequenceLstmLayer::CreateWorkload().

GetBackendId()

virtual const BackendId & GetBackendId ( ) const

pure virtual

Implemented in ClWorkloadFactory, NeonWorkloadFactory, RefWorkloadFactory, TosaRefWorkloadFactory, and SampleDynamicWorkloadFactory.

IsLayerSupported() [1/4]

bool IsLayerSupported ( const BackendId & backendId, const IConnectableLayer & layer, Optional< DataType > dataType, std::string & outReasonIfUnsupported )

static

Definition at line 1629 of file WorkloadFactory.cpp.

{
    return IsLayerConfigurationSupported(backendId, connectableLayer, dataType, outReasonIfUnsupported);
}

Referenced by armnn::AttemptBackendAssignment(), ClWorkloadFactory::IsLayerSupported(), ClWorkloadFactory::IsLayerSupported(), NeonWorkloadFactory::IsLayerSupported(), NeonWorkloadFactory::IsLayerSupported(), RefWorkloadFactory::IsLayerSupported(), RefWorkloadFactory::IsLayerSupported(), TosaRefWorkloadFactory::IsLayerSupported(), and TosaRefWorkloadFactory::IsLayerSupported().

IsLayerSupported() [2/4]

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

static

Definition at line 1658 of file WorkloadFactory.cpp.

{
    return IsLayerConfigurationSupported(backendId,
                                         connectableLayer,
                                         dataType,
                                         outReasonIfUnsupported,
                                         modelOptions);
}

IsLayerSupported() [3/4]

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

static

Definition at line 1637 of file WorkloadFactory.cpp.

{
    auto layer = PolymorphicDowncast<const Layer*>(&connectableLayer);
    return IsLayerConfigurationSupported(layer->GetBackendId(), connectableLayer, dataType, outReasonIfUnsupported);
}

References armnn::PolymorphicDowncast().

IsLayerSupported() [4/4]

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

static

Definition at line 1645 of file WorkloadFactory.cpp.

{
    auto layer = PolymorphicDowncast<const Layer*>(&connectableLayer);
    return IsLayerConfigurationSupported(layer->GetBackendId(),
                                         connectableLayer,
                                         dataType,
                                         outReasonIfUnsupported,
                                         modelOptions);
}

References armnn::PolymorphicDowncast().

SupportsSubTensors()

virtual bool SupportsSubTensors ( ) const

pure virtual

Implemented in ClWorkloadFactory, NeonWorkloadFactory, RefWorkloadFactory, TosaRefWorkloadFactory, WorkloadFactoryBase, and SampleDynamicWorkloadFactory.

---

The documentation for this class was generated from the following files:

• include/armnn/backends/WorkloadFactory.hpp
• src/backends/backendsCommon/WorkloadFactory.cpp