ClWorkloadFactory Class Reference

#include <ClWorkloadFactory.hpp>

Inheritance diagram for ClWorkloadFactory:

Public Member Functions
	ClWorkloadFactory (const std::shared_ptr< ClMemoryManager > &memoryManager)
	ClWorkloadFactory (const std::shared_ptr< ClMemoryManager > &memoryManager, const IBackendInternal::IBackendSpecificModelContextPtr &modelContextPtr)
void	AfterWorkloadsCreated () override
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 >	CreateAbs (const AbsQueueDescriptor &descriptor, const WorkloadInfo &info) 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 >	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 >	CreateConvertFp16ToFp32 (const ConvertFp16ToFp32QueueDescriptor &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 >	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 >	CreateEqual (const EqualQueueDescriptor &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 >	CreateGreater (const GreaterQueueDescriptor &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 >	CreateMerger (const MergerQueueDescriptor &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 >	CreateQuantizedLstm (const QuantizedLstmQueueDescriptor &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 >	CreateResizeBilinear (const ResizeBilinearQueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload >	CreateRsqrt (const RsqrtQueueDescriptor &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
Public Member Functions inherited from WorkloadFactoryBase
std::unique_ptr< IWorkload >	CreateFakeQuantization (const FakeQuantizationQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload >	CreateMerge (const MergeQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload >	CreateSwitch (const SwitchQueueDescriptor &, const WorkloadInfo &) const override
Public Member Functions inherited from IWorkloadFactory
virtual	~IWorkloadFactory ()
virtual std::unique_ptr< IWorkload >	CreateConvertBf16ToFp32 (const ConvertBf16ToFp32QueueDescriptor &descriptor, const WorkloadInfo &info) const
virtual std::unique_ptr< IWorkload >	CreateConvertFp32ToBf16 (const ConvertFp32ToBf16QueueDescriptor &descriptor, const WorkloadInfo &info) const
virtual std::unique_ptr< IWorkload >	CreateLogicalUnary (const ElementwiseUnaryQueueDescriptor &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 21 of file ClWorkloadFactory.hpp.

Constructor & Destructor Documentation

ClWorkloadFactory() [1/2]

ClWorkloadFactory

(

const std::shared_ptr< ClMemoryManager > &

memoryManager

)

Definition at line 138 of file ClWorkloadFactory.cpp.

    : m_MemoryManager(memoryManager), m_ModelContextPtr(IBackendInternal::IBackendSpecificModelContextPtr{})
{
    InitializeCLCompileContext();
}

ClWorkloadFactory() [2/2]

ClWorkloadFactory	(	const std::shared_ptr< ClMemoryManager > &	memoryManager,
		const IBackendInternal::IBackendSpecificModelContextPtr &	modelContextPtr
	)

Definition at line 144 of file ClWorkloadFactory.cpp.

    : m_MemoryManager(memoryManager), m_ModelContextPtr(modelContextPtr)
{
    InitializeCLCompileContext();
}

Member Function Documentation

AfterWorkloadsCreated()

void AfterWorkloadsCreated ( )

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 63 of file ClWorkloadFactory.cpp.

References CHECK_LOCATION, ClContextDeserializer::Deserialize(), armnn::Error, ClBackendModelContext::GetCachedNetworkFilePath(), armnn::info, ClContextSerializer::SaveSerializedToStream(), ClContextSerializer::Serialize(), and armnn::WrapClError().

{
    if(m_ModelContextPtr)
    {
        auto modelOptions = dynamic_cast<ClBackendModelContext*>(m_ModelContextPtr.get());
        if (modelOptions->SaveCachedNetwork())
        {
            // Save map to a filepath provided in ModelOptions
            auto filePath = modelOptions->GetCachedNetworkFilePath();
            if (filePath != "" && fs::exists(filePath) && fs::is_regular_file(filePath))
            {
                // Serialize ClContext to the file specified
                ClContextSerializer serializer;
                serializer.Serialize(m_CLCompileContext);
                std::ofstream file(filePath, std::ios::out | std::ios::binary);
                serializer.SaveSerializedToStream(file);
            }
        }
    }
}

CreateAbs()

std::unique_ptr< IWorkload > CreateAbs ( const AbsQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 197 of file ClWorkloadFactory.cpp.

References armnn::Abs, ClWorkloadFactory::CreateElementwiseUnary(), armnn::IgnoreUnused(), and QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    IgnoreUnused(descriptor);

    ElementwiseUnaryQueueDescriptor elementwiseUnaryDescriptor;
    elementwiseUnaryDescriptor.m_Parameters = ElementwiseUnaryDescriptor(UnaryOperation::Abs);

    return CreateElementwiseUnary(elementwiseUnaryDescriptor, info);
}

CreateActivation()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 208 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClActivationWorkload>(descriptor, info, m_CLCompileContext);
}

CreateAddition()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 214 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClAdditionWorkload>(descriptor, info, m_CLCompileContext);
}

CreateArgMinMax()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 220 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

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

CreateBatchNormalization()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 226 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

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

CreateBatchToSpaceNd()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 233 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClBatchToSpaceNdWorkload>(descriptor, info, m_CLCompileContext);
}

CreateComparison()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 239 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::CreateEqual(), ClWorkloadFactory::CreateGreater(), and ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClComparisonWorkload>(descriptor, info, m_CLCompileContext);
}

CreateConcat()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 245 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::CreateMerger(), and ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClConcatWorkload>(descriptor, info, m_CLCompileContext);
}

CreateConstant()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 251 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClConstantWorkload>(descriptor, info, m_CLCompileContext);
}

CreateConvertFp16ToFp32()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 257 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClConvertFp16ToFp32Workload>(descriptor, info, m_CLCompileContext);
}

CreateConvertFp32ToFp16()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 264 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClConvertFp32ToFp16Workload>(descriptor, info, m_CLCompileContext);
}

CreateConvolution2d()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 271 of file ClWorkloadFactory.cpp.

References armnn::info, and ClBackendModelContext::IsFastMathEnabled().

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    bool isFastMathEnabled = false;
    if (m_ModelContextPtr)
    {
        if (m_ModelContextPtr.get() != nullptr)
        {
            auto modelOptions = dynamic_cast<ClBackendModelContext*>(m_ModelContextPtr.get());
            if (modelOptions)
            {
                isFastMathEnabled = modelOptions->IsFastMathEnabled();
            }
        }
    }
    return MakeWorkload<ClConvolution2dWorkload>(descriptor,
                                                 info,
                                                 m_MemoryManager->GetIntraLayerManager(),
                                                 m_CLCompileContext,
                                                 isFastMathEnabled);
}

CreateDebug()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 293 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

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

CreateDepthToSpace()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 299 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClDepthToSpaceWorkload>(descriptor, info, m_CLCompileContext);
}

CreateDepthwiseConvolution2d()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 305 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClDepthwiseConvolutionWorkload>(descriptor, info, m_CLCompileContext);
}

CreateDequantize()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 312 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClDequantizeWorkload>(descriptor, info, m_CLCompileContext);
}

CreateDetectionPostProcess()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 318 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

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

CreateDivision()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 325 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

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

CreateElementwiseUnary()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 331 of file ClWorkloadFactory.cpp.

References armnn::Abs, armnn::Exp, armnn::info, armnn::LogicalNot, QueueDescriptor::m_Inputs, ElementwiseUnaryDescriptor::m_Operation, QueueDescriptor::m_Outputs, QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, armnn::Neg, and armnn::Rsqrt.

Referenced by ClWorkloadFactory::CreateAbs(), ClWorkloadFactory::CreateRsqrt(), and ClWorkloadFactory::SupportsSubTensors().

{
    switch(descriptor.m_Parameters.m_Operation)
    {
        case UnaryOperation::Abs:
        {
            AbsQueueDescriptor absQueueDescriptor;
            absQueueDescriptor.m_Inputs  = descriptor.m_Inputs;
            absQueueDescriptor.m_Outputs = descriptor.m_Outputs;

            return  std::make_unique<ClAbsWorkload>(absQueueDescriptor, info, m_CLCompileContext);
        }
        case UnaryOperation::Exp:
            return std::make_unique<ClExpWorkload>(descriptor, info, m_CLCompileContext);
        case UnaryOperation::Neg:
            return std::make_unique<ClNegWorkload>(descriptor, info, m_CLCompileContext);
        case UnaryOperation::Rsqrt:
        {
            RsqrtQueueDescriptor rsqrtQueueDescriptor;
            rsqrtQueueDescriptor.m_Inputs  = descriptor.m_Inputs;
            rsqrtQueueDescriptor.m_Outputs = descriptor.m_Outputs;

            return std::make_unique<ClRsqrtWorkload>(rsqrtQueueDescriptor, info, m_CLCompileContext);
        }
        case UnaryOperation::LogicalNot:
            return std::make_unique<ClLogicalNotWorkload>(descriptor, info, m_CLCompileContext);
        default:
            return nullptr;
    }
}

CreateEqual()

std::unique_ptr< IWorkload > CreateEqual ( const EqualQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 363 of file ClWorkloadFactory.cpp.

References ClWorkloadFactory::CreateComparison(), armnn::Equal, armnn::IgnoreUnused(), and QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    IgnoreUnused(descriptor);

    ComparisonQueueDescriptor comparisonDescriptor;
    comparisonDescriptor.m_Parameters = ComparisonDescriptor(ComparisonOperation::Equal);

    return CreateComparison(comparisonDescriptor, info);
}

CreateFill()

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

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 374 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

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

CreateFloor()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 380 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

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

CreateFullyConnected()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 386 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClFullyConnectedWorkload>(descriptor,
                                                  info,
                                                  m_MemoryManager->GetIntraLayerManager(),
                                                  m_CLCompileContext);
}

CreateGather()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 395 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClGatherWorkload>(descriptor, info, m_CLCompileContext);
}

CreateGreater()

std::unique_ptr< IWorkload > CreateGreater ( const GreaterQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 401 of file ClWorkloadFactory.cpp.

References ClWorkloadFactory::CreateComparison(), armnn::Greater, armnn::IgnoreUnused(), and QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    IgnoreUnused(descriptor);

    ComparisonQueueDescriptor comparisonDescriptor;
    comparisonDescriptor.m_Parameters = ComparisonDescriptor(ComparisonOperation::Greater);

    return CreateComparison(comparisonDescriptor, info);
}

CreateInput()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 412 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

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

CreateInstanceNormalization()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 418 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClInstanceNormalizationWorkload>(descriptor, info, m_CLCompileContext);
}

CreateL2Normalization()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 425 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

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

CreateLogicalBinary()

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

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 431 of file ClWorkloadFactory.cpp.

References armnn::info, armnn::LogicalAnd, armnn::LogicalOr, LogicalBinaryDescriptor::m_Operation, and QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    switch(descriptor.m_Parameters.m_Operation)
    {
        case LogicalBinaryOperation::LogicalAnd:
            return std::make_unique<ClLogicalAndWorkload>(descriptor, info, m_CLCompileContext);
        case LogicalBinaryOperation::LogicalOr:
            return std::make_unique<ClLogicalOrWorkload>(descriptor, info, m_CLCompileContext);
        default:
            return nullptr;
    }
}

CreateLogSoftmax()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 445 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClLogSoftmaxWorkload>(descriptor,
                                              info,
                                              m_MemoryManager->GetIntraLayerManager(),
                                              m_CLCompileContext);
}

CreateLstm()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 454 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

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

CreateMaximum()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 460 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClMaximumWorkload>(descriptor, info, m_CLCompileContext);
}

CreateMean()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 466 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClMeanWorkload>(descriptor, info, m_CLCompileContext);
}

CreateMemCopy()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 472 of file ClWorkloadFactory.cpp.

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

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    if (descriptor.m_Inputs.empty() || !descriptor.m_Inputs[0])
    {
        throw InvalidArgumentException("ClWorkloadFactory: Invalid null input for MemCopy workload");
    }

    return MakeWorkload<CopyMemGenericWorkload>(descriptor, info);
}

CreateMemImport()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 483 of file ClWorkloadFactory.cpp.

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

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    if (descriptor.m_Inputs.empty() || !descriptor.m_Inputs[0])
    {
        throw InvalidArgumentException("ClWorkloadFactory: Invalid null input for MemImport workload");
    }

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

CreateMerger()

std::unique_ptr< IWorkload > CreateMerger ( const MergerQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 494 of file ClWorkloadFactory.cpp.

References ClWorkloadFactory::CreateConcat().

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return CreateConcat(descriptor, info);
}

CreateMinimum()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 500 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClMinimumWorkload>(descriptor, info, m_CLCompileContext);
}

CreateMultiplication()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 506 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClMultiplicationWorkload>(descriptor, info, m_CLCompileContext);
}

CreateNormalization()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 512 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

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

CreateOutput()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 518 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

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

CreatePad()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 524 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClPadWorkload>(descriptor, info, m_CLCompileContext);
}

CreatePermute()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 530 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClPermuteWorkload>(descriptor, info, m_CLCompileContext);
}

CreatePooling2d()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 536 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClPooling2dWorkload>(descriptor, info, m_CLCompileContext);
}

CreatePreCompiled()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 542 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

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

CreatePrelu()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 548 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClPreluWorkload>(descriptor, info, m_CLCompileContext);
}

CreateQLstm()

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

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 554 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

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

CreateQuantize()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 560 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClQuantizeWorkload>(descriptor, info, m_CLCompileContext);
}

CreateQuantizedLstm()

std::unique_ptr< IWorkload > CreateQuantizedLstm ( const QuantizedLstmQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 566 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClQuantizedLstmWorkload>(descriptor, info, m_CLCompileContext);
}

CreateRank()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 572 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

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

CreateReduce()

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

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 578 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

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

CreateReshape()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 584 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClReshapeWorkload>(descriptor, info, m_CLCompileContext);
}

CreateResize()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 590 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::CreateResizeBilinear(), and ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClResizeWorkload>(descriptor, info, m_CLCompileContext);
}

CreateResizeBilinear()

std::unique_ptr< IWorkload > CreateResizeBilinear ( const ResizeBilinearQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from IWorkloadFactory.

Definition at line 596 of file ClWorkloadFactory.cpp.

References armnn::Bilinear, ClWorkloadFactory::CreateResize(), ResizeBilinearDescriptor::m_DataLayout, ResizeDescriptor::m_DataLayout, QueueDescriptor::m_Inputs, ResizeDescriptor::m_Method, QueueDescriptor::m_Outputs, QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, ResizeBilinearDescriptor::m_TargetHeight, ResizeDescriptor::m_TargetHeight, ResizeBilinearDescriptor::m_TargetWidth, and ResizeDescriptor::m_TargetWidth.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    ResizeQueueDescriptor resizeDescriptor;
    resizeDescriptor.m_Inputs  = descriptor.m_Inputs;
    resizeDescriptor.m_Outputs = descriptor.m_Outputs;

    resizeDescriptor.m_Parameters.m_Method       = ResizeMethod::Bilinear;
    resizeDescriptor.m_Parameters.m_DataLayout   = descriptor.m_Parameters.m_DataLayout;
    resizeDescriptor.m_Parameters.m_TargetHeight = descriptor.m_Parameters.m_TargetHeight;
    resizeDescriptor.m_Parameters.m_TargetWidth  = descriptor.m_Parameters.m_TargetWidth;

    return CreateResize(resizeDescriptor, info);
}

CreateRsqrt()

std::unique_ptr< IWorkload > CreateRsqrt ( const RsqrtQueueDescriptor &  descriptor, const WorkloadInfo &  info  ) const

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 611 of file ClWorkloadFactory.cpp.

References ClWorkloadFactory::CreateElementwiseUnary(), armnn::IgnoreUnused(), QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, and armnn::Rsqrt.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    IgnoreUnused(descriptor);

    ElementwiseUnaryQueueDescriptor elementwiseUnaryDescriptor;
    elementwiseUnaryDescriptor.m_Parameters = ElementwiseUnaryDescriptor(UnaryOperation::Rsqrt);

    return CreateElementwiseUnary(elementwiseUnaryDescriptor, info);
}

CreateSlice()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 622 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClSliceWorkload>(descriptor, info, m_CLCompileContext);
}

CreateSoftmax()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 628 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return std::make_unique<ClSoftmaxWorkload>(descriptor,
                                               info,
                                               m_MemoryManager->GetIntraLayerManager(),
                                               m_CLCompileContext);
}

CreateSpaceToBatchNd()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 637 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClSpaceToBatchNdWorkload>(descriptor, info, m_CLCompileContext);
}

CreateSpaceToDepth()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 643 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClSpaceToDepthWorkload>(descriptor, info, m_CLCompileContext);
}

CreateSplitter()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 649 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClSplitterWorkload>(descriptor, info, m_CLCompileContext);
}

CreateStack()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 655 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClStackWorkload>(descriptor, info, m_CLCompileContext);
}

CreateStridedSlice()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 661 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClStridedSliceWorkload>(descriptor, info, m_CLCompileContext);
}

CreateSubTensorHandle()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 172 of file ClWorkloadFactory.cpp.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    arm_compute::Coordinates coords;
    arm_compute::TensorShape shape = armcomputetensorutils::BuildArmComputeTensorShape(subTensorShape);

    coords.set_num_dimensions(subTensorShape.GetNumDimensions());
    for (unsigned int i = 0; i < subTensorShape.GetNumDimensions(); i++)
    {
        // Arm compute indexes tensor coords in reverse order.
        unsigned int revertedIndex = subTensorShape.GetNumDimensions() - i - 1;
        coords.set(i, armnn::numeric_cast<int>(subTensorOrigin[revertedIndex]));
    }

    const arm_compute::TensorShape parentShape = armcomputetensorutils::BuildArmComputeTensorShape(parent.GetShape());
    if (!::arm_compute::error_on_invalid_subtensor(__func__, __FILE__, __LINE__, parentShape, coords, shape))
    {
        return nullptr;
    }

    return std::make_unique<ClSubTensorHandle>(
        PolymorphicDowncast<IClTensorHandle*>(&parent), shape, coords);
}

CreateSubtraction()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 667 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClSubtractionWorkload>(descriptor, info, m_CLCompileContext);
}

CreateTensorHandle() [1/2]

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 151 of file ClWorkloadFactory.cpp.

References armnn::IgnoreUnused().

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    IgnoreUnused(IsMemoryManaged);
    std::unique_ptr<ClTensorHandle> tensorHandle = std::make_unique<ClTensorHandle>(tensorInfo);
    tensorHandle->SetMemoryGroup(m_MemoryManager->GetInterLayerMemoryGroup());

    return tensorHandle;
}

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 161 of file ClWorkloadFactory.cpp.

References armnn::IgnoreUnused().

{
    IgnoreUnused(IsMemoryManaged);
    std::unique_ptr<ClTensorHandle> tensorHandle = std::make_unique<ClTensorHandle>(tensorInfo, dataLayout);
    tensorHandle->SetMemoryGroup(m_MemoryManager->GetInterLayerMemoryGroup());

    return tensorHandle;
}

CreateTranspose()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 673 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClTransposeWorkload>(descriptor, info, m_CLCompileContext);
}

CreateTransposeConvolution2d()

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

overridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 679 of file ClWorkloadFactory.cpp.

References armnn::info.

Referenced by ClWorkloadFactory::SupportsSubTensors().

{
    return MakeWorkload<ClTransposeConvolution2dWorkload>(descriptor,
                                                          info,
                                                          m_MemoryManager->GetIntraLayerManager(),
                                                          m_CLCompileContext);
}

GetBackendId()

const BackendId & GetBackendId ( ) const

overridevirtual

Implements IWorkloadFactory.

Definition at line 58 of file ClWorkloadFactory.cpp.

{
    return s_Id;
}

IsLayerSupported() [1/2]

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

static

Definition at line 43 of file ClWorkloadFactory.cpp.

References IWorkloadFactory::IsLayerSupported().

{
    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 50 of file ClWorkloadFactory.cpp.

References IWorkloadFactory::IsLayerSupported().

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

SupportsSubTensors()

bool SupportsSubTensors ( ) const

inlineoverridevirtual

Reimplemented from WorkloadFactoryBase.

Definition at line 42 of file ClWorkloadFactory.hpp.

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

{ return true; }

---

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

• src/backends/cl/ClWorkloadFactory.hpp
• src/backends/cl/ClWorkloadFactory.cpp