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	SupportsInPlaceComputation () const
virtual bool	SupportsSubTensors () const =0
virtual bool	SupportsMapUnmap () const
virtual MemorySourceFlags	GetExportFlags () const
virtual MemorySourceFlags	GetImportFlags () const
virtual std::vector< Capability >	GetCapabilities (const IConnectableLayer *layer, const IConnectableLayer *connectedLayer, CapabilityClass capabilityClass)

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 46 of file ITensorHandleFactory.hpp.

Member Typedef Documentation

FactoryId

using FactoryId = std::string

Definition at line 49 of file ITensorHandleFactory.hpp.

Constructor & Destructor Documentation

~ITensorHandleFactory()

virtual ~ITensorHandleFactory ( )

inlinevirtual

Some TensorHandleFactory decisions are deferred to run-time.

Definition at line 53 of file ITensorHandleFactory.hpp.

{}

Member Function Documentation

CreateSubTensorHandle()

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

pure virtual

Implemented in MockTensorHandleFactory, RefTensorHandleFactory, TosaRefTensorHandleFactory, GpuFsaTensorHandleFactory, NeonTensorHandleFactory, ClImportTensorHandleFactory, ClTensorHandleFactory, and SampleDynamicTensorHandleFactory.

CreateTensorHandle() [1/4]

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

pure virtual

Implemented in NeonTensorHandleFactory, ClImportTensorHandleFactory, MockTensorHandleFactory, ClTensorHandleFactory, RefTensorHandleFactory, TosaRefTensorHandleFactory, GpuFsaTensorHandleFactory, and SampleDynamicTensorHandleFactory.

Referenced by ITensorHandleFactory::CreateTensorHandle(), OutputHandler::CreateTensorHandles(), LoadedNetwork::CreateWorkingMemHandle(), LoadedNetwork::ImportInputs(), and LoadedNetwork::ImportOutputs().

CreateTensorHandle() [2/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 NeonTensorHandleFactory, ClImportTensorHandleFactory, MockTensorHandleFactory, ClTensorHandleFactory, RefTensorHandleFactory, TosaRefTensorHandleFactory, GpuFsaTensorHandleFactory, and SampleDynamicTensorHandleFactory.

Definition at line 67 of file ITensorHandleFactory.hpp.

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

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

CreateTensorHandle() [3/4]

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

pure virtual

Implemented in NeonTensorHandleFactory, ClImportTensorHandleFactory, MockTensorHandleFactory, ClTensorHandleFactory, RefTensorHandleFactory, TosaRefTensorHandleFactory, GpuFsaTensorHandleFactory, and SampleDynamicTensorHandleFactory.

CreateTensorHandle() [4/4]

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

inlinevirtual

Reimplemented in NeonTensorHandleFactory, ClImportTensorHandleFactory, MockTensorHandleFactory, ClTensorHandleFactory, RefTensorHandleFactory, TosaRefTensorHandleFactory, GpuFsaTensorHandleFactory, and SampleDynamicTensorHandleFactory.

Definition at line 74 of file ITensorHandleFactory.hpp.

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

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

GetCapabilities()

virtual std::vector<Capability> GetCapabilities ( const IConnectableLayer *  layer, const IConnectableLayer *  connectedLayer, CapabilityClass  capabilityClass  )

inlinevirtual

Reimplemented in NeonTensorHandleFactory, and ClImportTensorHandleFactory.

Definition at line 93 of file ITensorHandleFactory.hpp.

    {
        IgnoreUnused(layer);
        IgnoreUnused(connectedLayer);
        IgnoreUnused(capabilityClass);
        return std::vector<Capability>();
    }

References armnn::IgnoreUnused().

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

GetExportFlags()

virtual MemorySourceFlags GetExportFlags ( ) const

inlinevirtual

Reimplemented in NeonTensorHandleFactory, ClImportTensorHandleFactory, MockTensorHandleFactory, ClTensorHandleFactory, RefTensorHandleFactory, TosaRefTensorHandleFactory, SampleDynamicTensorHandleFactory, and GpuFsaTensorHandleFactory.

Definition at line 90 of file ITensorHandleFactory.hpp.

{ return 0; }

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

GetId()

virtual const FactoryId& GetId ( ) const

pure virtual

Implemented in NeonTensorHandleFactory, ClImportTensorHandleFactory, MockTensorHandleFactory, ClTensorHandleFactory, RefTensorHandleFactory, TosaRefTensorHandleFactory, SampleDynamicTensorHandleFactory, and GpuFsaTensorHandleFactory.

GetImportFlags()

virtual MemorySourceFlags GetImportFlags ( ) const

inlinevirtual

Reimplemented in NeonTensorHandleFactory, ClImportTensorHandleFactory, MockTensorHandleFactory, ClTensorHandleFactory, RefTensorHandleFactory, TosaRefTensorHandleFactory, SampleDynamicTensorHandleFactory, and GpuFsaTensorHandleFactory.

Definition at line 91 of file ITensorHandleFactory.hpp.

{ return 0; }

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

SupportsInPlaceComputation()

virtual bool SupportsInPlaceComputation ( ) const

inlinevirtual

Reimplemented in NeonTensorHandleFactory.

Definition at line 84 of file ITensorHandleFactory.hpp.

{ return false; }

SupportsMapUnmap()

virtual bool SupportsMapUnmap ( ) const

inlinevirtual

Reimplemented in ClImportTensorHandleFactory.

Definition at line 88 of file ITensorHandleFactory.hpp.

{ return true; }

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

SupportsSubTensors()

virtual bool SupportsSubTensors ( ) const

pure virtual

Implemented in NeonTensorHandleFactory, ClImportTensorHandleFactory, MockTensorHandleFactory, ClTensorHandleFactory, RefTensorHandleFactory, TosaRefTensorHandleFactory, SampleDynamicTensorHandleFactory, and GpuFsaTensorHandleFactory.

Member Data Documentation

DeferredFactoryId

const ITensorHandleFactory::FactoryId DeferredFactoryId = "armnn_deferred_factory"

static

Use the workload factory to create the tensor handle.

Definition at line 51 of file ITensorHandleFactory.hpp.

Referenced by armnn::CalculateSlotOptionForOutput().

LegacyFactoryId

const ITensorHandleFactory::FactoryId LegacyFactoryId = "armnn_legacy_factory"

static

Definition at line 50 of file ITensorHandleFactory.hpp.

Referenced by Graph::AddCompatibilityLayers(), armnn::CalculateEdgeStrategy(), armnn::CalculateSlotOption(), armnn::CalculateSlotOptionForInput(), SplitterLayer::CreateTensorHandles(), ConcatLayer::CreateTensorHandles(), Layer::CreateTensorHandles(), LoadedNetwork::CreateWorkingMemHandle(), 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