20.11
|
#include <NeonTensorHandleFactory.hpp>
Public Member Functions | |
NeonTensorHandleFactory (std::weak_ptr< NeonMemoryManager > 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=true) const override |
const FactoryId & | GetId () const override |
bool | SupportsInPlaceComputation () const override |
bool | SupportsSubTensors () const override |
MemorySourceFlags | GetExportFlags () const override |
MemorySourceFlags | GetImportFlags () const override |
std::vector< Capability > | GetCapabilities (const IConnectableLayer *layer, const IConnectableLayer *connectedLayer, CapabilityClass capabilityClass) override |
![]() | |
virtual | ~ITensorHandleFactory () |
Some TensorHandleFactory decisions are deferred to run-time. More... | |
virtual bool | SupportsMapUnmap () const final |
Static Public Member Functions | |
static const FactoryId & | GetIdStatic () |
Additional Inherited Members | |
![]() | |
using | FactoryId = std::string |
![]() | |
static const FactoryId | LegacyFactoryId = "armnn_legacy_factory" |
static const FactoryId | DeferredFactoryId = "armnn_deferred_factory" |
Use the workload factory to create the tensor handle. More... | |
Definition at line 34 of file NeonTensorHandleFactory.hpp.
|
inline |
Definition at line 37 of file NeonTensorHandleFactory.hpp.
References NeonTensorHandleFactory::CreateSubTensorHandle(), NeonTensorHandleFactory::CreateTensorHandle(), NeonTensorHandleFactory::GetCapabilities(), NeonTensorHandleFactory::GetExportFlags(), NeonTensorHandleFactory::GetId(), NeonTensorHandleFactory::GetIdStatic(), NeonTensorHandleFactory::GetImportFlags(), NeonTensorHandleFactory::SupportsInPlaceComputation(), and NeonTensorHandleFactory::SupportsSubTensors().
|
overridevirtual |
Implements ITensorHandleFactory.
Definition at line 20 of file NeonTensorHandleFactory.cpp.
Referenced by NeonTensorHandleFactory::NeonTensorHandleFactory().
|
overridevirtual |
Implements ITensorHandleFactory.
Definition at line 47 of file NeonTensorHandleFactory.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), NeonTensorHandleFactory::CreateTensorHandle(), and NeonTensorHandleFactory::NeonTensorHandleFactory().
|
overridevirtual |
Implements ITensorHandleFactory.
Definition at line 52 of file NeonTensorHandleFactory.cpp.
References NeonTensorHandleFactory::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 58 of file NeonTensorHandleFactory.cpp.
References NeonTensorHandleFactory::GetImportFlags().
|
overridevirtual |
Reimplemented from ITensorHandleFactory.
Definition at line 73 of file NeonTensorHandleFactory.cpp.
References NeonTensorHandleFactory::GetImportFlags().
|
overridevirtual |
Reimplemented from ITensorHandleFactory.
Definition at line 120 of file NeonTensorHandleFactory.cpp.
References armnn::IgnoreUnused(), armnn::PaddingRequired, and armnn::paddingRequiredLayers.
Referenced by NeonTensorHandleFactory::NeonTensorHandleFactory().
|
overridevirtual |
Reimplemented from ITensorHandleFactory.
Definition at line 110 of file NeonTensorHandleFactory.cpp.
Referenced by NeonTensorHandleFactory::NeonTensorHandleFactory().
|
overridevirtual |
Implements ITensorHandleFactory.
Definition at line 95 of file NeonTensorHandleFactory.cpp.
References NeonTensorHandleFactory::GetIdStatic().
Referenced by NeonTensorHandleFactory::NeonTensorHandleFactory().
|
static |
Definition at line 89 of file NeonTensorHandleFactory.cpp.
References armnn::NeonTensorHandleFactoryId().
Referenced by NeonBackend::GetHandleFactoryPreferences(), NeonTensorHandleFactory::GetId(), and NeonTensorHandleFactory::NeonTensorHandleFactory().
|
overridevirtual |
Reimplemented from ITensorHandleFactory.
Definition at line 115 of file NeonTensorHandleFactory.cpp.
Referenced by NeonTensorHandleFactory::CreateTensorHandle(), and NeonTensorHandleFactory::NeonTensorHandleFactory().
|
overridevirtual |
Reimplemented from ITensorHandleFactory.
Definition at line 100 of file NeonTensorHandleFactory.cpp.
Referenced by NeonTensorHandleFactory::NeonTensorHandleFactory().
|
overridevirtual |
Implements ITensorHandleFactory.
Definition at line 105 of file NeonTensorHandleFactory.cpp.
Referenced by NeonTensorHandleFactory::NeonTensorHandleFactory().