20.08
|
#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 | SupportsSubTensors () const override |
MemorySourceFlags | GetExportFlags () const override |
MemorySourceFlags | GetImportFlags () const override |
std::vector< Capability > | GetCapabilities (const IConnectableLayer *layer, const IConnectableLayer *connectedLayer, CapabilityClass capabilityClass) override |
Public Member Functions inherited from ITensorHandleFactory | |
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 | |
Public Types inherited from ITensorHandleFactory | |
using | FactoryId = std::string |
Static Public Attributes inherited from ITensorHandleFactory | |
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 35 of file NeonTensorHandleFactory.hpp.
|
inline |
Definition at line 38 of file NeonTensorHandleFactory.hpp.
References NeonTensorHandleFactory::CreateSubTensorHandle(), NeonTensorHandleFactory::CreateTensorHandle(), NeonTensorHandleFactory::GetCapabilities(), NeonTensorHandleFactory::GetExportFlags(), NeonTensorHandleFactory::GetId(), NeonTensorHandleFactory::GetIdStatic(), NeonTensorHandleFactory::GetImportFlags(), and NeonTensorHandleFactory::SupportsSubTensors().
|
overridevirtual |
Implements ITensorHandleFactory.
Definition at line 19 of file NeonTensorHandleFactory.cpp.
Referenced by NeonTensorHandleFactory::NeonTensorHandleFactory().
|
overridevirtual |
Implements ITensorHandleFactory.
Definition at line 46 of file NeonTensorHandleFactory.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), NeonTensorHandleFactory::CreateTensorHandle(), and NeonTensorHandleFactory::NeonTensorHandleFactory().
|
overridevirtual |
Implements ITensorHandleFactory.
Definition at line 51 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 57 of file NeonTensorHandleFactory.cpp.
References NeonTensorHandleFactory::GetImportFlags().
|
overridevirtual |
Reimplemented from ITensorHandleFactory.
Definition at line 72 of file NeonTensorHandleFactory.cpp.
References NeonTensorHandleFactory::GetImportFlags().
|
overridevirtual |
Reimplemented from ITensorHandleFactory.
Definition at line 114 of file NeonTensorHandleFactory.cpp.
References armnn::IgnoreUnused(), armnn::PaddingRequired, and armnn::paddingRequiredLayers.
Referenced by NeonTensorHandleFactory::NeonTensorHandleFactory().
|
overridevirtual |
Reimplemented from ITensorHandleFactory.
Definition at line 104 of file NeonTensorHandleFactory.cpp.
Referenced by NeonTensorHandleFactory::NeonTensorHandleFactory().
|
overridevirtual |
Implements ITensorHandleFactory.
Definition at line 94 of file NeonTensorHandleFactory.cpp.
References NeonTensorHandleFactory::GetIdStatic().
Referenced by NeonTensorHandleFactory::NeonTensorHandleFactory().
|
static |
Definition at line 88 of file NeonTensorHandleFactory.cpp.
References armnn::NeonTensorHandleFactoryId().
Referenced by NeonTensorHandleFactory::GetId(), and NeonTensorHandleFactory::NeonTensorHandleFactory().
|
overridevirtual |
Reimplemented from ITensorHandleFactory.
Definition at line 109 of file NeonTensorHandleFactory.cpp.
Referenced by NeonTensorHandleFactory::CreateTensorHandle(), and NeonTensorHandleFactory::NeonTensorHandleFactory().
|
overridevirtual |
Implements ITensorHandleFactory.
Definition at line 99 of file NeonTensorHandleFactory.cpp.
Referenced by NeonTensorHandleFactory::NeonTensorHandleFactory().