20.11
|
#include <ClTensorHandleFactory.hpp>
Public Member Functions | |
ClTensorHandleFactory (std::shared_ptr< ClMemoryManager > mgr) | |
std::unique_ptr< ITensorHandle > | CreateSubTensorHandle (ITensorHandle &parent, const TensorShape &subTensorShape, const unsigned int *subTensorOrigin) const override |
std::unique_ptr< ITensorHandle > | CreateTensorHandle (const TensorInfo &tensorInfo) const override |
std::unique_ptr< ITensorHandle > | CreateTensorHandle (const TensorInfo &tensorInfo, DataLayout dataLayout) const override |
std::unique_ptr< ITensorHandle > | CreateTensorHandle (const TensorInfo &tensorInfo, const bool IsMemoryManaged) const override |
Utility Functions for backends which require TensorHandles to have unmanaged memory. More... | |
std::unique_ptr< ITensorHandle > | CreateTensorHandle (const TensorInfo &tensorInfo, DataLayout dataLayout, const bool IsMemoryManaged) const override |
const FactoryId & | GetId () const override |
bool | SupportsSubTensors () const override |
MemorySourceFlags | GetExportFlags () const override |
MemorySourceFlags | GetImportFlags () const override |
![]() | |
virtual | ~ITensorHandleFactory () |
Some TensorHandleFactory decisions are deferred to run-time. More... | |
virtual bool | SupportsInPlaceComputation () const |
virtual bool | SupportsMapUnmap () const final |
virtual std::vector< Capability > | GetCapabilities (const IConnectableLayer *layer, const IConnectableLayer *connectedLayer, CapabilityClass capabilityClass) |
Static Public Member Functions | |
static const FactoryId & | GetIdStatic () |
Static Public Attributes | |
static const FactoryId | m_Id |
![]() | |
static const FactoryId | LegacyFactoryId = "armnn_legacy_factory" |
static const FactoryId | DeferredFactoryId = "armnn_deferred_factory" |
Use the workload factory to create the tensor handle. More... | |
Additional Inherited Members | |
![]() | |
using | FactoryId = std::string |
Definition at line 17 of file ClTensorHandleFactory.hpp.
|
inline |
Definition at line 21 of file ClTensorHandleFactory.hpp.
References ClTensorHandleFactory::CreateSubTensorHandle(), ClTensorHandleFactory::CreateTensorHandle(), ClTensorHandleFactory::GetExportFlags(), ClTensorHandleFactory::GetId(), ClTensorHandleFactory::GetIdStatic(), ClTensorHandleFactory::GetImportFlags(), and ClTensorHandleFactory::SupportsSubTensors().
|
overridevirtual |
Implements ITensorHandleFactory.
Definition at line 23 of file ClTensorHandleFactory.cpp.
Referenced by ClTensorHandleFactory::ClTensorHandleFactory().
|
overridevirtual |
Implements ITensorHandleFactory.
Definition at line 61 of file ClTensorHandleFactory.cpp.
Referenced by ClTensorHandleFactory::ClTensorHandleFactory(), and ClTensorHandleFactory::CreateTensorHandle().
|
overridevirtual |
Implements ITensorHandleFactory.
Definition at line 66 of file ClTensorHandleFactory.cpp.
References ClTensorHandleFactory::CreateTensorHandle().
|
overridevirtual |
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 from ITensorHandleFactory.
Definition at line 72 of file ClTensorHandleFactory.cpp.
References ARMNN_LOG, and armnn::warning.
|
overridevirtual |
Reimplemented from ITensorHandleFactory.
Definition at line 84 of file ClTensorHandleFactory.cpp.
References ARMNN_LOG, and armnn::warning.
|
overridevirtual |
Reimplemented from ITensorHandleFactory.
Definition at line 113 of file ClTensorHandleFactory.cpp.
Referenced by ClTensorHandleFactory::ClTensorHandleFactory().
|
overridevirtual |
Implements ITensorHandleFactory.
Definition at line 103 of file ClTensorHandleFactory.cpp.
References ClTensorHandleFactory::GetIdStatic().
Referenced by ClTensorHandleFactory::ClTensorHandleFactory().
|
static |
Definition at line 97 of file ClTensorHandleFactory.cpp.
References armnn::ClTensorHandleFactoryId().
Referenced by ClTensorHandleFactory::ClTensorHandleFactory(), ClBackend::GetHandleFactoryPreferences(), and ClTensorHandleFactory::GetId().
|
overridevirtual |
Reimplemented from ITensorHandleFactory.
Definition at line 118 of file ClTensorHandleFactory.cpp.
Referenced by ClTensorHandleFactory::ClTensorHandleFactory().
|
overridevirtual |
Implements ITensorHandleFactory.
Definition at line 108 of file ClTensorHandleFactory.cpp.
Referenced by ClTensorHandleFactory::ClTensorHandleFactory().
|
static |
Definition at line 19 of file ClTensorHandleFactory.hpp.