20.05
|
#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=true) 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 |
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 16 of file NeonTensorHandleFactory.hpp.
|
inline |
Definition at line 19 of file NeonTensorHandleFactory.hpp.
References NeonTensorHandleFactory::CreateSubTensorHandle(), NeonTensorHandleFactory::CreateTensorHandle(), NeonTensorHandleFactory::GetExportFlags(), NeonTensorHandleFactory::GetId(), NeonTensorHandleFactory::GetIdStatic(), NeonTensorHandleFactory::GetImportFlags(), and NeonTensorHandleFactory::SupportsSubTensors().
|
overridevirtual |
Implements ITensorHandleFactory.
Definition at line 17 of file NeonTensorHandleFactory.cpp.
Referenced by NeonTensorHandleFactory::NeonTensorHandleFactory().
|
overridevirtual |
Implements ITensorHandleFactory.
Definition at line 43 of file NeonTensorHandleFactory.cpp.
Referenced by NeonTensorHandleFactory::CreateTensorHandle(), and NeonTensorHandleFactory::NeonTensorHandleFactory().
|
overridevirtual |
Implements ITensorHandleFactory.
Definition at line 48 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 54 of file NeonTensorHandleFactory.cpp.
References NeonTensorHandleFactory::GetImportFlags().
|
overridevirtual |
Reimplemented from ITensorHandleFactory.
Definition at line 69 of file NeonTensorHandleFactory.cpp.
References NeonTensorHandleFactory::GetImportFlags().
|
overridevirtual |
Reimplemented from ITensorHandleFactory.
Definition at line 101 of file NeonTensorHandleFactory.cpp.
Referenced by NeonTensorHandleFactory::NeonTensorHandleFactory().
|
overridevirtual |
Implements ITensorHandleFactory.
Definition at line 91 of file NeonTensorHandleFactory.cpp.
References NeonTensorHandleFactory::GetIdStatic().
Referenced by NeonTensorHandleFactory::NeonTensorHandleFactory().
|
static |
Definition at line 85 of file NeonTensorHandleFactory.cpp.
References armnn::NeonTensorHandleFactoryId().
Referenced by NeonTensorHandleFactory::GetId(), and NeonTensorHandleFactory::NeonTensorHandleFactory().
|
overridevirtual |
Reimplemented from ITensorHandleFactory.
Definition at line 106 of file NeonTensorHandleFactory.cpp.
Referenced by NeonTensorHandleFactory::CreateTensorHandle(), and NeonTensorHandleFactory::NeonTensorHandleFactory().
|
overridevirtual |
Implements ITensorHandleFactory.
Definition at line 96 of file NeonTensorHandleFactory.cpp.
Referenced by NeonTensorHandleFactory::NeonTensorHandleFactory().