ITensorHandleFactory Class Reference

#include <ITensorHandleFactory.hpp>

Inheritance diagram for ITensorHandleFactory:

Public Types
using	FactoryId = std::string

Public Member Functions
virtual	~ITensorHandleFactory ()
	Some TensorHandleFactory decisions are deferred to run-time. More...
virtual std::unique_ptr< ITensorHandle >	CreateSubTensorHandle (ITensorHandle &parent, TensorShape const &subTensorShape, unsigned int const *subTensorOrigin) const =0
virtual std::unique_ptr< ITensorHandle >	CreateTensorHandle (const TensorInfo &tensorInfo) const =0
virtual std::unique_ptr< ITensorHandle >	CreateTensorHandle (const TensorInfo &tensorInfo, DataLayout dataLayout) const =0
virtual std::unique_ptr< ITensorHandle >	CreateTensorHandle (const TensorInfo &tensorInfo, const bool IsMemoryManaged) const
	Utility Functions for backends which require TensorHandles to have unmanaged memory. More...
virtual std::unique_ptr< ITensorHandle >	CreateTensorHandle (const TensorInfo &tensorInfo, DataLayout dataLayout, const bool IsMemoryManaged) const
virtual const FactoryId &	GetId () const =0
virtual bool	SupportsSubTensors () const =0
virtual bool	SupportsMapUnmap () const final
virtual MemorySourceFlags	GetExportFlags () const
virtual MemorySourceFlags	GetImportFlags () const

Static Public Attributes
static const FactoryId	LegacyFactoryId = "armnn_legacy_factory"
static const FactoryId	DeferredFactoryId = "armnn_deferred_factory"
	Use the workload factory to create the tensor handle. More...

Detailed Description

Definition at line 18 of file ITensorHandleFactory.hpp.

Member Typedef Documentation

FactoryId

using FactoryId = std::string

Definition at line 21 of file ITensorHandleFactory.hpp.

Constructor & Destructor Documentation

~ITensorHandleFactory()

virtual ~ITensorHandleFactory ( )

inlinevirtual

Some TensorHandleFactory decisions are deferred to run-time.

Definition at line 25 of file ITensorHandleFactory.hpp.

References ITensorHandleFactory::CreateSubTensorHandle(), and ITensorHandleFactory::CreateTensorHandle().

{}

Member Function Documentation

CreateSubTensorHandle()

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

pure virtual

Implemented in ClTensorHandleFactory, RefTensorHandleFactory, and NeonTensorHandleFactory.

Referenced by ITensorHandleFactory::~ITensorHandleFactory().

CreateTensorHandle() [1/4]

virtual std::unique_ptr<ITensorHandle> CreateTensorHandle ( const TensorInfo &  tensorInfo ) const

pure virtual

Implemented in ClTensorHandleFactory, RefTensorHandleFactory, and NeonTensorHandleFactory.

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

CreateTensorHandle() [2/4]

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

pure virtual

Implemented in ClTensorHandleFactory, RefTensorHandleFactory, and NeonTensorHandleFactory.

CreateTensorHandle() [3/4]

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

inlinevirtual

Utility Functions for backends which require TensorHandles to have unmanaged memory.

These should be overloaded if required to facilitate direct import of input tensors and direct export of output tensors.

Reimplemented in ClTensorHandleFactory, and NeonTensorHandleFactory.

Definition at line 39 of file ITensorHandleFactory.hpp.

References ITensorHandleFactory::CreateTensorHandle(), and armnn::IgnoreUnused().

    {
        IgnoreUnused(IsMemoryManaged);
        return CreateTensorHandle(tensorInfo);
    }

CreateTensorHandle() [4/4]

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

inlinevirtual

Reimplemented in ClTensorHandleFactory, and NeonTensorHandleFactory.

Definition at line 46 of file ITensorHandleFactory.hpp.

References ITensorHandleFactory::CreateTensorHandle(), ITensorHandleFactory::GetId(), armnn::IgnoreUnused(), and ITensorHandleFactory::SupportsSubTensors().

    {
        IgnoreUnused(IsMemoryManaged);
        return CreateTensorHandle(tensorInfo, dataLayout);
    }

GetExportFlags()

virtual MemorySourceFlags GetExportFlags ( ) const

inlinevirtual

Reimplemented in ClTensorHandleFactory, NeonTensorHandleFactory, and RefTensorHandleFactory.

Definition at line 60 of file ITensorHandleFactory.hpp.

Referenced by armnn::CalculateEdgeStrategy(), and armnn::RequiresCopy().

{ return 0; }

GetId()

virtual const FactoryId& GetId ( ) const

pure virtual

Implemented in ClTensorHandleFactory, NeonTensorHandleFactory, and RefTensorHandleFactory.

Referenced by ITensorHandleFactory::CreateTensorHandle().

GetImportFlags()

virtual MemorySourceFlags GetImportFlags ( ) const

inlinevirtual

Reimplemented in ClTensorHandleFactory, NeonTensorHandleFactory, and RefTensorHandleFactory.

Definition at line 61 of file ITensorHandleFactory.hpp.

Referenced by armnn::CalculateEdgeStrategy(), armnn::CalculateSlotOptionForInput(), and armnn::RequiresCopy().

{ return 0; }

SupportsMapUnmap()

virtual bool SupportsMapUnmap ( ) const

inlinefinalvirtual

Definition at line 58 of file ITensorHandleFactory.hpp.

Referenced by armnn::CalculateEdgeStrategy(), armnn::CalculateSlotOption(), and armnn::CalculateSlotOptionForInput().

{ return true; }

SupportsSubTensors()

virtual bool SupportsSubTensors ( ) const

pure virtual

Implemented in ClTensorHandleFactory, NeonTensorHandleFactory, and RefTensorHandleFactory.

Referenced by ITensorHandleFactory::CreateTensorHandle().

Member Data Documentation

DeferredFactoryId

const ITensorHandleFactory::FactoryId DeferredFactoryId = "armnn_deferred_factory"

static

Use the workload factory to create the tensor handle.

Definition at line 23 of file ITensorHandleFactory.hpp.

Referenced by armnn::CalculateSlotOptionForOutput().

LegacyFactoryId

const ITensorHandleFactory::FactoryId LegacyFactoryId = "armnn_legacy_factory"

static

Definition at line 22 of file ITensorHandleFactory.hpp.

Referenced by Graph::AddCompatibilityLayers(), armnn::CalculateEdgeStrategy(), armnn::CalculateSlotOption(), armnn::CalculateSlotOptionForInput(), ConcatLayer::CreateTensorHandles(), SplitterLayer::CreateTensorHandles(), Layer::CreateTensorHandles(), IBackendInternal::GetBackwardCompatibleFavoriteHandleFactory(), and armnn::SelectTensorHandleStrategy().

---

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

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