SampleDynamicWorkloadFactory Class Reference

#include <SampleDynamicWorkloadFactory.hpp>

Inheritance diagram for SampleDynamicWorkloadFactory:

Public Member Functions
	SampleDynamicWorkloadFactory (const std::shared_ptr< SampleMemoryManager > &memoryManager)
	SampleDynamicWorkloadFactory ()
	~SampleDynamicWorkloadFactory ()
const armnn::BackendId &	GetBackendId () const override
bool	SupportsSubTensors () const override
std::unique_ptr< armnn::ITensorHandle >	CreateSubTensorHandle (armnn::ITensorHandle &parent, armnn::TensorShape const &subTensorShape, unsigned int const *subTensorOrigin) const override
std::unique_ptr< armnn::ITensorHandle >	CreateTensorHandle (const armnn::TensorInfo &tensorInfo, const bool IsMemoryManaged=true) const override
std::unique_ptr< armnn::ITensorHandle >	CreateTensorHandle (const armnn::TensorInfo &tensorInfo, armnn::DataLayout dataLayout, const bool IsMemoryManaged=true) const override
std::unique_ptr< armnn::IWorkload >	CreateAddition (const armnn::AdditionQueueDescriptor &descriptor, const armnn::WorkloadInfo &info) const override
std::unique_ptr< armnn::IWorkload >	CreateInput (const armnn::InputQueueDescriptor &descriptor, const armnn::WorkloadInfo &info) const override
std::unique_ptr< armnn::IWorkload >	CreateOutput (const armnn::OutputQueueDescriptor &descriptor, const armnn::WorkloadInfo &info) const override
Public Member Functions inherited from IWorkloadFactory
virtual	~IWorkloadFactory ()
virtual void	AfterWorkloadsCreated ()
virtual std::unique_ptr< IWorkload >	CreateWorkload (LayerType type, const QueueDescriptor &descriptor, const WorkloadInfo &info) const

Static Public Member Functions
static bool	IsLayerSupported (const armnn::IConnectableLayer &layer, armnn::Optional< armnn::DataType > dataType, std::string &outReasonIfUnsupported)
Static 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 16 of file SampleDynamicWorkloadFactory.hpp.

Constructor & Destructor Documentation

SampleDynamicWorkloadFactory() [1/2]

SampleDynamicWorkloadFactory ( const std::shared_ptr< SampleMemoryManager > &  memoryManager )

explicit

Definition at line 22 of file SampleDynamicWorkloadFactory.cpp.

    : m_MemoryManager(memoryManager)
{
}

SampleDynamicWorkloadFactory() [2/2]

SampleDynamicWorkloadFactory

(

)

Definition at line 27 of file SampleDynamicWorkloadFactory.cpp.

    : m_MemoryManager(new SampleMemoryManager())
{
}

~SampleDynamicWorkloadFactory()

~SampleDynamicWorkloadFactory ( )

inline

Definition at line 22 of file SampleDynamicWorkloadFactory.hpp.

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

{}

Member Function Documentation

CreateAddition()

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

override

Definition at line 59 of file SampleDynamicWorkloadFactory.cpp.

Referenced by SampleDynamicWorkloadFactory::CreateSubTensorHandle().

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

CreateInput()

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

overridevirtual

Implements IWorkloadFactory.

Definition at line 66 of file SampleDynamicWorkloadFactory.cpp.

Referenced by SampleDynamicWorkloadFactory::CreateSubTensorHandle().

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

CreateOutput()

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

override

Definition at line 73 of file SampleDynamicWorkloadFactory.cpp.

Referenced by SampleDynamicWorkloadFactory::CreateSubTensorHandle().

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

CreateSubTensorHandle()

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

inlineoverridevirtual

Implements IWorkloadFactory.

Definition at line 33 of file SampleDynamicWorkloadFactory.hpp.

References ARMNN_DEPRECATED_MSG, SampleDynamicWorkloadFactory::CreateAddition(), SampleDynamicWorkloadFactory::CreateInput(), SampleDynamicWorkloadFactory::CreateOutput(), SampleDynamicWorkloadFactory::CreateTensorHandle(), and armnn::IgnoreUnused().

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

CreateTensorHandle() [1/2]

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

overridevirtual

Implements IWorkloadFactory.

Definition at line 44 of file SampleDynamicWorkloadFactory.cpp.

Referenced by SampleDynamicWorkloadFactory::CreateSubTensorHandle().

{
    return std::make_unique<armnn::ScopedTensorHandle>(tensorInfo);
}

CreateTensorHandle() [2/2]

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

overridevirtual

Implements IWorkloadFactory.

Definition at line 51 of file SampleDynamicWorkloadFactory.cpp.

{
    return std::make_unique<armnn::ScopedTensorHandle>(tensorInfo);
}

GetBackendId()

const armnn::BackendId & GetBackendId ( ) const

overridevirtual

Implements IWorkloadFactory.

Definition at line 32 of file SampleDynamicWorkloadFactory.cpp.

Referenced by SampleDynamicWorkloadFactory::~SampleDynamicWorkloadFactory().

{
    return s_Id;
}

IsLayerSupported()

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

static

Definition at line 37 of file SampleDynamicWorkloadFactory.cpp.

References armnn::IsLayerSupported().

Referenced by SampleDynamicWorkloadFactory::~SampleDynamicWorkloadFactory().

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

SupportsSubTensors()

bool SupportsSubTensors ( ) const

inlineoverridevirtual

Implements IWorkloadFactory.

Definition at line 30 of file SampleDynamicWorkloadFactory.hpp.

References ARMNN_DEPRECATED_MSG.

{ return false; }

---

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

• src/dynamic/sample/SampleDynamicWorkloadFactory.hpp
• src/dynamic/sample/SampleDynamicWorkloadFactory.cpp