RefWorkloadFactory Class Reference

#include <RefWorkloadFactory.hpp>

Inheritance diagram for RefWorkloadFactory:

Public Member Functions
	RefWorkloadFactory (const std::shared_ptr< RefMemoryManager > &memoryManager)
	RefWorkloadFactory ()
	~RefWorkloadFactory ()
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 >	CreateActivation (const ActivationQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateAddition (const AdditionQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateArgMinMax (const ArgMinMaxQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateBatchNormalization (const BatchNormalizationQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateBatchToSpaceNd (const BatchToSpaceNdQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateCast (const CastQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateChannelShuffle (const ChannelShuffleQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateComparison (const ComparisonQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateConcat (const ConcatQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateConstant (const ConstantQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateConvertBf16ToFp32 (const ConvertBf16ToFp32QueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateConvertFp16ToFp32 (const ConvertFp16ToFp32QueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateConvertFp32ToBf16 (const ConvertFp32ToBf16QueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateConvertFp32ToFp16 (const ConvertFp32ToFp16QueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateConvolution2d (const Convolution2dQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateConvolution3d (const Convolution3dQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateDebug (const DebugQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateDepthToSpace (const DepthToSpaceQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateDepthwiseConvolution2d (const DepthwiseConvolution2dQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateDequantize (const DequantizeQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateDetectionPostProcess (const DetectionPostProcessQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateDivision (const DivisionQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateElementwiseUnary (const ElementwiseUnaryQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateFakeQuantization (const FakeQuantizationQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateFill (const FillQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateFloor (const FloorQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateFullyConnected (const FullyConnectedQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateGather (const GatherQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateInput (const InputQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateInstanceNormalization (const InstanceNormalizationQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateL2Normalization (const L2NormalizationQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateLogicalBinary (const LogicalBinaryQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateLogSoftmax (const LogSoftmaxQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateLstm (const LstmQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateMaximum (const MaximumQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateMean (const MeanQueueDescriptor &descriptor, const WorkloadInfo &Info) const override
std::unique_ptr< IWorkload >	CreateMemCopy (const MemCopyQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateMemImport (const MemImportQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateMinimum (const MinimumQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateMultiplication (const MultiplicationQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateNormalization (const NormalizationQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateOutput (const OutputQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreatePad (const PadQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreatePermute (const PermuteQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreatePooling2d (const Pooling2dQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreatePreCompiled (const PreCompiledQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreatePrelu (const PreluQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateQLstm (const QLstmQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateQuantize (const QuantizeQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateRank (const RankQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateReduce (const ReduceQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateReshape (const ReshapeQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateResize (const ResizeQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateShape (const ShapeQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateSlice (const SliceQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateSoftmax (const SoftmaxQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateSpaceToBatchNd (const SpaceToBatchNdQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateSpaceToDepth (const SpaceToDepthQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateSplitter (const SplitterQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateStack (const StackQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateStridedSlice (const StridedSliceQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateSubtraction (const SubtractionQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateTranspose (const TransposeQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateTransposeConvolution2d (const TransposeConvolution2dQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateUnidirectionalSequenceLstm (const UnidirectionalSequenceLstmQueueDescriptor &descriptor, const WorkloadInfo &info) const override
Public Member Functions inherited from IWorkloadFactory
virtual	~IWorkloadFactory ()
virtual void	AfterWorkloadsCreated ()
virtual std::unique_ptr< IWorkload >	CreateLogicalUnary (const ElementwiseUnaryQueueDescriptor &descriptor, const WorkloadInfo &Info) const
virtual std::unique_ptr< IWorkload >	CreateMerge (const MergeQueueDescriptor &descriptor, const WorkloadInfo &info) const
virtual std::unique_ptr< IWorkload >	CreateQuantizedLstm (const QuantizedLstmQueueDescriptor &descriptor, const WorkloadInfo &info) const
virtual std::unique_ptr< IWorkload >	CreateSwitch (const SwitchQueueDescriptor &descriptor, const WorkloadInfo &Info) const

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 30 of file RefWorkloadFactory.hpp.

Constructor & Destructor Documentation

RefWorkloadFactory() [1/2]

RefWorkloadFactory ( const std::shared_ptr< RefMemoryManager > &  memoryManager )

explicit

Definition at line 83 of file RefWorkloadFactory.cpp.

    : m_MemoryManager(memoryManager)
{
}

RefWorkloadFactory() [2/2]

RefWorkloadFactory

(

)

Definition at line 88 of file RefWorkloadFactory.cpp.

    : m_MemoryManager(new RefMemoryManager())
{
}

~RefWorkloadFactory()

~RefWorkloadFactory ( )

inline

Definition at line 36 of file RefWorkloadFactory.hpp.

References RefWorkloadFactory::GetBackendId(), and RefWorkloadFactory::IsLayerSupported().

{}

Member Function Documentation

CreateActivation()

std::unique_ptr< IWorkload > CreateActivation ( const ActivationQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 144 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefActivationWorkload>(descriptor, info);
}

CreateAddition()

std::unique_ptr< IWorkload > CreateAddition ( const AdditionQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 150 of file RefWorkloadFactory.cpp.

References armnn::info, WorkloadInfo::m_InputTensorInfos, and armnn::Signed32.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32)
    {
        return std::make_unique<RefAdditionWorkload<int32_t>>(descriptor, info);
    }
    else
    {
        return std::make_unique<RefAdditionWorkload<float>>(descriptor, info);
    }
}

CreateArgMinMax()

std::unique_ptr< IWorkload > CreateArgMinMax ( const ArgMinMaxQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 163 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefArgMinMaxWorkload>(descriptor, info);
}

CreateBatchNormalization()

std::unique_ptr< IWorkload > CreateBatchNormalization ( const BatchNormalizationQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 169 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefBatchNormalizationWorkload>(descriptor, info);
}

CreateBatchToSpaceNd()

std::unique_ptr< IWorkload > CreateBatchToSpaceNd ( const BatchToSpaceNdQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 176 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefBatchToSpaceNdWorkload>(descriptor, info);
}

CreateCast()

std::unique_ptr< IWorkload > CreateCast ( const CastQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 182 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefCastWorkload>(descriptor, info);
}

CreateChannelShuffle()

std::unique_ptr< IWorkload > CreateChannelShuffle ( const ChannelShuffleQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 188 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefChannelShuffleWorkload>(descriptor,info);
}

CreateComparison()

std::unique_ptr< IWorkload > CreateComparison ( const ComparisonQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 194 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefComparisonWorkload>(descriptor, info);
}

CreateConcat()

std::unique_ptr< IWorkload > CreateConcat ( const ConcatQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 200 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefConcatWorkload>(descriptor, info);
}

CreateConstant()

std::unique_ptr< IWorkload > CreateConstant ( const ConstantQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 206 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefConstantWorkload>(descriptor, info);
}

CreateConvertBf16ToFp32()

std::unique_ptr< IWorkload > CreateConvertBf16ToFp32 ( const ConvertBf16ToFp32QueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 212 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefConvertBf16ToFp32Workload>(descriptor, info);
}

CreateConvertFp16ToFp32()

std::unique_ptr< IWorkload > CreateConvertFp16ToFp32 ( const ConvertFp16ToFp32QueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 219 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefConvertFp16ToFp32Workload>(descriptor, info);
}

CreateConvertFp32ToBf16()

std::unique_ptr< IWorkload > CreateConvertFp32ToBf16 ( const ConvertFp32ToBf16QueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 226 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefConvertFp32ToBf16Workload>(descriptor, info);
}

CreateConvertFp32ToFp16()

std::unique_ptr< IWorkload > CreateConvertFp32ToFp16 ( const ConvertFp32ToFp16QueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 233 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefConvertFp32ToFp16Workload>(descriptor, info);
}

CreateConvolution2d()

std::unique_ptr< IWorkload > CreateConvolution2d ( const Convolution2dQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 240 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefConvolution2dWorkload>(descriptor, info);
}

CreateConvolution3d()

std::unique_ptr< IWorkload > CreateConvolution3d ( const Convolution3dQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 246 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefConvolution3dWorkload>(descriptor, info);
}

CreateDebug()

std::unique_ptr< IWorkload > CreateDebug ( const DebugQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 252 of file RefWorkloadFactory.cpp.

References armnn::info, armnn::IsBFloat16(), armnn::IsFloat16(), armnn::IsQAsymmS8(), armnn::IsQAsymmU8(), armnn::IsQSymmS16(), armnn::IsQSymmS8(), and armnn::IsSigned32().

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    if (IsBFloat16(info))
    {
        return std::make_unique<RefDebugBFloat16Workload>(descriptor, info);
    }
    if (IsFloat16(info))
    {
        return std::make_unique<RefDebugFloat16Workload>(descriptor, info);
    }
    if (IsQSymmS16(info))
    {
        return std::make_unique<RefDebugQSymmS16Workload>(descriptor, info);
    }
    if (IsQSymmS8(info))
    {
        return std::make_unique<RefDebugQSymmS8Workload>(descriptor, info);
    }
    if (IsQAsymmU8(info))
    {
        return std::make_unique<RefDebugQAsymmU8Workload>(descriptor, info);
    }
    if (IsQAsymmS8(info))
    {
        return std::make_unique<RefDebugQAsymmS8Workload>(descriptor, info);
    }
    if (IsSigned32(info))
    {
        return std::make_unique<RefDebugSigned32Workload>(descriptor, info);
    }

    return MakeWorkload<RefDebugFloat32Workload, RefDebugQAsymmU8Workload>(descriptor, info);
}

CreateDepthToSpace()

std::unique_ptr< IWorkload > CreateDepthToSpace ( const DepthToSpaceQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 287 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefDepthToSpaceWorkload>(descriptor, info);
}

CreateDepthwiseConvolution2d()

std::unique_ptr< IWorkload > CreateDepthwiseConvolution2d ( const DepthwiseConvolution2dQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 293 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefDepthwiseConvolution2dWorkload>(descriptor, info);
}

CreateDequantize()

std::unique_ptr< IWorkload > CreateDequantize ( const DequantizeQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 300 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefDequantizeWorkload>(descriptor, info);
}

CreateDetectionPostProcess()

std::unique_ptr< IWorkload > CreateDetectionPostProcess ( const DetectionPostProcessQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 306 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefDetectionPostProcessWorkload>(descriptor, info);
}

CreateDivision()

std::unique_ptr< IWorkload > CreateDivision ( const DivisionQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 313 of file RefWorkloadFactory.cpp.

References armnn::info, WorkloadInfo::m_InputTensorInfos, and armnn::Signed32.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32)
    {
        return std::make_unique<RefDivisionWorkload<int32_t>>(descriptor, info);
    }
    else
    {
        return std::make_unique<RefDivisionWorkload<float>>(descriptor, info);
    }
}

CreateElementwiseUnary()

std::unique_ptr< IWorkload > CreateElementwiseUnary ( const ElementwiseUnaryQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 326 of file RefWorkloadFactory.cpp.

References armnn::info, armnn::LogicalNot, ElementwiseUnaryDescriptor::m_Operation, and QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    if (descriptor.m_Parameters.m_Operation == UnaryOperation::LogicalNot)
    {
        return std::make_unique<RefLogicalUnaryWorkload>(descriptor, info);
    }
    return std::make_unique<RefElementwiseUnaryWorkload>(descriptor, info);
}

CreateFakeQuantization()

std::unique_ptr< IWorkload > CreateFakeQuantization ( const FakeQuantizationQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 336 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return MakeWorkload<RefFakeQuantizationFloat32Workload, NullWorkload>(descriptor, info);
}

CreateFill()

std::unique_ptr< IWorkload > CreateFill ( const FillQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 342 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefFillWorkload>(descriptor, info);
}

CreateFloor()

std::unique_ptr< IWorkload > CreateFloor ( const FloorQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 348 of file RefWorkloadFactory.cpp.

References armnn::info, armnn::IsQuantizedType(), and WorkloadInfo::m_InputTensorInfos.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    if(IsQuantizedType(info.m_InputTensorInfos[0].GetDataType()))
    {
        return nullptr;
    }
    else
    {
        return std::make_unique<RefFloorWorkload>(descriptor, info);
    }
}

CreateFullyConnected()

std::unique_ptr< IWorkload > CreateFullyConnected ( const FullyConnectedQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 361 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefFullyConnectedWorkload>(descriptor, info);
}

CreateGather()

std::unique_ptr< IWorkload > CreateGather ( const GatherQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 368 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefGatherWorkload>(descriptor, info);
}

CreateInput()

std::unique_ptr< IWorkload > CreateInput ( const InputQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Implements IWorkloadFactory.

Definition at line 374 of file RefWorkloadFactory.cpp.

References armnn::info, WorkloadInfo::m_InputTensorInfos, and WorkloadInfo::m_OutputTensorInfos.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    if (info.m_InputTensorInfos.empty() )
    {
        throw InvalidArgumentException("RefWorkloadFactory::CreateInput: Input cannot be zero length");
    }
    if (info.m_OutputTensorInfos.empty())
    {
        throw InvalidArgumentException("RefWorkloadFactory::CreateInput: Output cannot be zero length");
    }

    if (info.m_InputTensorInfos[0].GetNumBytes() != info.m_OutputTensorInfos[0].GetNumBytes())
    {
        throw InvalidArgumentException("RefWorkloadFactory::CreateInput: data input and output differ in byte count.");
    }

    return std::make_unique<CopyMemGenericWorkload>(descriptor, info);
}

CreateInstanceNormalization()

std::unique_ptr< IWorkload > CreateInstanceNormalization ( const InstanceNormalizationQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 394 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefInstanceNormalizationWorkload>(descriptor, info);
}

CreateL2Normalization()

std::unique_ptr< IWorkload > CreateL2Normalization ( const L2NormalizationQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 401 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefL2NormalizationWorkload>(descriptor, info);
}

CreateLogicalBinary()

std::unique_ptr< IWorkload > CreateLogicalBinary ( const LogicalBinaryQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 407 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefLogicalBinaryWorkload>(descriptor, info);
}

CreateLogSoftmax()

std::unique_ptr< IWorkload > CreateLogSoftmax ( const LogSoftmaxQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 413 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefLogSoftmaxWorkload>(descriptor, info);
}

CreateLstm()

std::unique_ptr< IWorkload > CreateLstm ( const LstmQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 419 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefLstmWorkload>(descriptor, info);
}

CreateMaximum()

std::unique_ptr< IWorkload > CreateMaximum ( const MaximumQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 425 of file RefWorkloadFactory.cpp.

References armnn::info, WorkloadInfo::m_InputTensorInfos, and armnn::Signed32.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32)
    {
        return std::make_unique<RefMaximumWorkload<int32_t>>(descriptor, info);
    }
    else
    {
        return std::make_unique<RefMaximumWorkload<float>>(descriptor, info);
    }
}

CreateMean()

std::unique_ptr< IWorkload > CreateMean ( const MeanQueueDescriptor &  descriptor, const WorkloadInfo &  Info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 438 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return  std::make_unique<RefMeanWorkload>(descriptor, info);
}

CreateMemCopy()

std::unique_ptr< IWorkload > CreateMemCopy ( const MemCopyQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 444 of file RefWorkloadFactory.cpp.

References armnn::info, and QueueDescriptor::m_Inputs.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    if (descriptor.m_Inputs.empty())
    {
        throw InvalidArgumentException("RefWorkloadFactory: CreateMemCopy() expected an input tensor.");
    }
    return std::make_unique<CopyMemGenericWorkload>(descriptor, info);
}

CreateMemImport()

std::unique_ptr< IWorkload > CreateMemImport ( const MemImportQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 454 of file RefWorkloadFactory.cpp.

References armnn::info, and QueueDescriptor::m_Inputs.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    if (descriptor.m_Inputs.empty())
    {
        throw InvalidArgumentException("RefWorkloadFactory: CreateMemImport() expected an input tensor.");
    }
    return std::make_unique<ImportMemGenericWorkload>(descriptor, info);
}

CreateMinimum()

std::unique_ptr< IWorkload > CreateMinimum ( const MinimumQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 464 of file RefWorkloadFactory.cpp.

References armnn::info, WorkloadInfo::m_InputTensorInfos, and armnn::Signed32.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32)
    {
        return std::make_unique<RefMinimumWorkload<int32_t>>(descriptor, info);
    }
    else
    {
        return std::make_unique<RefMinimumWorkload<float>>(descriptor, info);
    }
}

CreateMultiplication()

std::unique_ptr< IWorkload > CreateMultiplication ( const MultiplicationQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 477 of file RefWorkloadFactory.cpp.

References armnn::info, WorkloadInfo::m_InputTensorInfos, and armnn::Signed32.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32)
    {
        return std::make_unique<RefMultiplicationWorkload<int32_t>>(descriptor, info);
    }
    else
    {
        return std::make_unique<RefMultiplicationWorkload<float>>(descriptor, info);
    }
}

CreateNormalization()

std::unique_ptr< IWorkload > CreateNormalization ( const NormalizationQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 490 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefNormalizationWorkload>(descriptor, info);
}

CreateOutput()

std::unique_ptr< IWorkload > CreateOutput ( const OutputQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 496 of file RefWorkloadFactory.cpp.

References armnn::info, WorkloadInfo::m_InputTensorInfos, and WorkloadInfo::m_OutputTensorInfos.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    if (info.m_InputTensorInfos.empty() )
    {
        throw InvalidArgumentException("RefWorkloadFactory::CreateOutput: Input cannot be zero length");
    }
    if (info.m_OutputTensorInfos.empty())
    {
        throw InvalidArgumentException("RefWorkloadFactory::CreateOutput: Output cannot be zero length");
    }
    if (info.m_InputTensorInfos[0].GetNumBytes() != info.m_OutputTensorInfos[0].GetNumBytes())
    {
        throw InvalidArgumentException("RefWorkloadFactory::CreateOutput: data input and output differ in byte count.");
    }

    return std::make_unique<CopyMemGenericWorkload>(descriptor, info);
}

CreatePad()

std::unique_ptr< IWorkload > CreatePad ( const PadQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 515 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefPadWorkload>(descriptor, info);
}

CreatePermute()

std::unique_ptr< IWorkload > CreatePermute ( const PermuteQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 521 of file RefWorkloadFactory.cpp.

References armnn::info, armnn::IsBFloat16(), armnn::IsQAsymmS8(), and armnn::IsQSymmS16().

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    if (IsQSymmS16(info))
    {
        return std::make_unique<RefPermuteQSymm16Workload>(descriptor, info);
    }
    else if (IsBFloat16(info))
    {
        return std::make_unique<RefPermuteBFloat16Workload>(descriptor, info);
    }
    else if (IsQAsymmS8(info))
    {
        return std::make_unique<RefPermuteQAsymmS8Workload>(descriptor, info);
    }
    return MakeWorkloadHelper<RefPermuteFloat16Workload, RefPermuteFloat32Workload, RefPermuteQAsymm8Workload,
        NullWorkload, NullWorkload, NullWorkload>(descriptor, info);
}

CreatePooling2d()

std::unique_ptr< IWorkload > CreatePooling2d ( const Pooling2dQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 540 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefPooling2dWorkload>(descriptor, info);
}

CreatePreCompiled()

std::unique_ptr< IWorkload > CreatePreCompiled ( const PreCompiledQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 546 of file RefWorkloadFactory.cpp.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return nullptr;
}

CreatePrelu()

std::unique_ptr< IWorkload > CreatePrelu ( const PreluQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 552 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefPreluWorkload>(descriptor, info);
}

CreateQLstm()

std::unique_ptr< IWorkload > CreateQLstm ( const QLstmQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 558 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefQLstmWorkload>(descriptor, info);
}

CreateQuantize()

std::unique_ptr< IWorkload > CreateQuantize ( const QuantizeQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 564 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefQuantizeWorkload>(descriptor, info);
}

CreateRank()

std::unique_ptr< IWorkload > CreateRank ( const RankQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 570 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefRankWorkload>(descriptor, info);
}

CreateReduce()

std::unique_ptr< IWorkload > CreateReduce ( const ReduceQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 576 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefReduceWorkload>(descriptor, info);
}

CreateReshape()

std::unique_ptr< IWorkload > CreateReshape ( const ReshapeQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 582 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefReshapeWorkload>(descriptor, info);
}

CreateResize()

std::unique_ptr< IWorkload > CreateResize ( const ResizeQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 588 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefResizeWorkload>(descriptor, info);
}

CreateShape()

std::unique_ptr< IWorkload > CreateShape ( const ShapeQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 594 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefShapeWorkload>(descriptor, info);
}

CreateSlice()

std::unique_ptr< IWorkload > CreateSlice ( const SliceQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 600 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefSliceWorkload>(descriptor, info);
}

CreateSoftmax()

std::unique_ptr< IWorkload > CreateSoftmax ( const SoftmaxQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 606 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefSoftmaxWorkload>(descriptor, info);
}

CreateSpaceToBatchNd()

std::unique_ptr< IWorkload > CreateSpaceToBatchNd ( const SpaceToBatchNdQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 612 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefSpaceToBatchNdWorkload>(descriptor, info);
}

CreateSpaceToDepth()

std::unique_ptr< IWorkload > CreateSpaceToDepth ( const SpaceToDepthQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 618 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefSpaceToDepthWorkload>(descriptor, info);
}

CreateSplitter()

std::unique_ptr< IWorkload > CreateSplitter ( const SplitterQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 624 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefSplitterWorkload>(descriptor, info);
}

CreateStack()

std::unique_ptr< IWorkload > CreateStack ( const StackQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 630 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefStackWorkload>(descriptor, info);
}

CreateStridedSlice()

std::unique_ptr< IWorkload > CreateStridedSlice ( const StridedSliceQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 636 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefStridedSliceWorkload>(descriptor, info);
}

CreateSubTensorHandle()

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

inlineoverridevirtual

Implements IWorkloadFactory.

Definition at line 52 of file RefWorkloadFactory.hpp.

References ARMNN_DEPRECATED_MSG, RefWorkloadFactory::CreateActivation(), RefWorkloadFactory::CreateAddition(), RefWorkloadFactory::CreateArgMinMax(), RefWorkloadFactory::CreateBatchNormalization(), RefWorkloadFactory::CreateBatchToSpaceNd(), RefWorkloadFactory::CreateCast(), RefWorkloadFactory::CreateChannelShuffle(), RefWorkloadFactory::CreateComparison(), RefWorkloadFactory::CreateConcat(), RefWorkloadFactory::CreateConstant(), RefWorkloadFactory::CreateConvertBf16ToFp32(), RefWorkloadFactory::CreateConvertFp16ToFp32(), RefWorkloadFactory::CreateConvertFp32ToBf16(), RefWorkloadFactory::CreateConvertFp32ToFp16(), RefWorkloadFactory::CreateConvolution2d(), RefWorkloadFactory::CreateConvolution3d(), RefWorkloadFactory::CreateDebug(), RefWorkloadFactory::CreateDepthToSpace(), RefWorkloadFactory::CreateDepthwiseConvolution2d(), RefWorkloadFactory::CreateDequantize(), RefWorkloadFactory::CreateDetectionPostProcess(), RefWorkloadFactory::CreateDivision(), RefWorkloadFactory::CreateElementwiseUnary(), RefWorkloadFactory::CreateFakeQuantization(), RefWorkloadFactory::CreateFill(), RefWorkloadFactory::CreateFloor(), RefWorkloadFactory::CreateFullyConnected(), RefWorkloadFactory::CreateGather(), RefWorkloadFactory::CreateInput(), RefWorkloadFactory::CreateInstanceNormalization(), RefWorkloadFactory::CreateL2Normalization(), RefWorkloadFactory::CreateLogicalBinary(), RefWorkloadFactory::CreateLogSoftmax(), RefWorkloadFactory::CreateLstm(), RefWorkloadFactory::CreateMaximum(), RefWorkloadFactory::CreateMean(), RefWorkloadFactory::CreateMemCopy(), RefWorkloadFactory::CreateMemImport(), RefWorkloadFactory::CreateMinimum(), RefWorkloadFactory::CreateMultiplication(), RefWorkloadFactory::CreateNormalization(), RefWorkloadFactory::CreateOutput(), RefWorkloadFactory::CreatePad(), RefWorkloadFactory::CreatePermute(), RefWorkloadFactory::CreatePooling2d(), RefWorkloadFactory::CreatePreCompiled(), RefWorkloadFactory::CreatePrelu(), RefWorkloadFactory::CreateQLstm(), RefWorkloadFactory::CreateQuantize(), RefWorkloadFactory::CreateRank(), RefWorkloadFactory::CreateReduce(), RefWorkloadFactory::CreateReshape(), RefWorkloadFactory::CreateResize(), RefWorkloadFactory::CreateShape(), RefWorkloadFactory::CreateSlice(), RefWorkloadFactory::CreateSoftmax(), RefWorkloadFactory::CreateSpaceToBatchNd(), RefWorkloadFactory::CreateSpaceToDepth(), RefWorkloadFactory::CreateSplitter(), RefWorkloadFactory::CreateStack(), RefWorkloadFactory::CreateStridedSlice(), RefWorkloadFactory::CreateSubtraction(), RefWorkloadFactory::CreateTensorHandle(), RefWorkloadFactory::CreateTranspose(), RefWorkloadFactory::CreateTransposeConvolution2d(), RefWorkloadFactory::CreateUnidirectionalSequenceLstm(), armnn::IgnoreUnused(), armnn::Info, and armnn::info.

    {
        IgnoreUnused(parent, subTensorShape, subTensorOrigin);
        return nullptr;
    }

CreateSubtraction()

std::unique_ptr< IWorkload > CreateSubtraction ( const SubtractionQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 642 of file RefWorkloadFactory.cpp.

References armnn::info, WorkloadInfo::m_InputTensorInfos, and armnn::Signed32.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32)
    {
        return std::make_unique<RefSubtractionWorkload<int32_t>>(descriptor, info);
    }
    else
    {
        return std::make_unique<RefSubtractionWorkload<float>>(descriptor, info);
    }
}

CreateTensorHandle() [1/2]

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

overridevirtual

Implements IWorkloadFactory.

Definition at line 113 of file RefWorkloadFactory.cpp.

References armnn::Malloc.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    if (isMemoryManaged)
    {
        return std::make_unique<RefTensorHandle>(tensorInfo, m_MemoryManager);
    }
    else
    {
        return std::make_unique<RefTensorHandle>(tensorInfo, static_cast<unsigned int>(MemorySource::Malloc));
    }
}

CreateTensorHandle() [2/2]

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

overridevirtual

Implements IWorkloadFactory.

Definition at line 126 of file RefWorkloadFactory.cpp.

References armnn::IgnoreUnused(), and armnn::Malloc.

{
    // For Ref it is okay to make the TensorHandle memory managed as it can also store a pointer
    // to unmanaged memory. This also ensures memory alignment.
    IgnoreUnused(isMemoryManaged, dataLayout);

    if (isMemoryManaged)
    {
        return std::make_unique<RefTensorHandle>(tensorInfo, m_MemoryManager);
    }
    else
    {
        return std::make_unique<RefTensorHandle>(tensorInfo, static_cast<unsigned int>(MemorySource::Malloc));
    }
}

CreateTranspose()

std::unique_ptr< IWorkload > CreateTranspose ( const TransposeQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 655 of file RefWorkloadFactory.cpp.

References armnn::info, armnn::IsBFloat16(), armnn::IsQAsymmS8(), and armnn::IsQSymmS16().

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    if (IsQSymmS16(info))
    {
        return std::make_unique<RefTransposeQSymm16Workload>(descriptor, info);
    }
    else if (IsBFloat16(info))
    {
        return std::make_unique<RefTransposeBFloat16Workload>(descriptor, info);
    }
    else if (IsQAsymmS8(info))
    {
        return std::make_unique<RefTransposeQAsymmS8Workload>(descriptor, info);
    }
    return MakeWorkloadHelper<RefTransposeFloat16Workload, RefTransposeFloat32Workload, RefTransposeQAsymm8Workload,
            NullWorkload, NullWorkload, NullWorkload>(descriptor, info);
}

CreateTransposeConvolution2d()

std::unique_ptr< IWorkload > CreateTransposeConvolution2d ( const TransposeConvolution2dQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 674 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefTransposeConvolution2dWorkload>(descriptor, info);
}

CreateUnidirectionalSequenceLstm()

std::unique_ptr< IWorkload > CreateUnidirectionalSequenceLstm ( const UnidirectionalSequenceLstmQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 681 of file RefWorkloadFactory.cpp.

References armnn::info.

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<RefUnidirectionalSequenceLstmWorkload>(descriptor, info);;
}

GetBackendId()

const BackendId & GetBackendId ( ) const

overridevirtual

Implements IWorkloadFactory.

Definition at line 93 of file RefWorkloadFactory.cpp.

Referenced by RefWorkloadFactory::~RefWorkloadFactory().

{
    return s_Id;
}

IsLayerSupported() [1/2]

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

static

Definition at line 98 of file RefWorkloadFactory.cpp.

References IWorkloadFactory::IsLayerSupported().

Referenced by RefWorkloadFactory::~RefWorkloadFactory().

{
    return IWorkloadFactory::IsLayerSupported(s_Id, layer, dataType, outReasonIfUnsupported);
}

IsLayerSupported() [2/2]

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

static

Definition at line 105 of file RefWorkloadFactory.cpp.

References IWorkloadFactory::IsLayerSupported().

{
    return IWorkloadFactory::IsLayerSupported(s_Id, layer, dataType, outReasonIfUnsupported, modelOptions);
}

SupportsSubTensors()

bool SupportsSubTensors ( ) const

inlineoverridevirtual

Implements IWorkloadFactory.

Definition at line 49 of file RefWorkloadFactory.hpp.

References ARMNN_DEPRECATED_MSG.

{ return false; }

---

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

• src/backends/reference/RefWorkloadFactory.hpp
• src/backends/reference/RefWorkloadFactory.cpp