20.02
|
#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 16 of file NeonTensorHandleFactory.cpp.
Referenced by NeonTensorHandleFactory::NeonTensorHandleFactory().
|
overridevirtual |
Implements ITensorHandleFactory.
Definition at line 42 of file NeonTensorHandleFactory.cpp.
Referenced by NeonTensorHandleFactory::CreateTensorHandle(), and NeonTensorHandleFactory::NeonTensorHandleFactory().
|
overridevirtual |
Implements ITensorHandleFactory.
Definition at line 47 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 53 of file NeonTensorHandleFactory.cpp.
References NeonTensorHandleFactory::GetImportFlags().
|
overridevirtual |
Reimplemented from ITensorHandleFactory.
Definition at line 68 of file NeonTensorHandleFactory.cpp.
References NeonTensorHandleFactory::GetImportFlags().
|
overridevirtual |
Reimplemented from ITensorHandleFactory.
Definition at line 100 of file NeonTensorHandleFactory.cpp.
Referenced by NeonTensorHandleFactory::NeonTensorHandleFactory().
|
overridevirtual |
Implements ITensorHandleFactory.
Definition at line 90 of file NeonTensorHandleFactory.cpp.
References NeonTensorHandleFactory::GetIdStatic().
Referenced by NeonTensorHandleFactory::NeonTensorHandleFactory().
|
static |
Definition at line 84 of file NeonTensorHandleFactory.cpp.
References armnn::NeonTensorHandleFactoryId().
Referenced by NeonTensorHandleFactory::GetId(), and NeonTensorHandleFactory::NeonTensorHandleFactory().
|
overridevirtual |
Reimplemented from ITensorHandleFactory.
Definition at line 105 of file NeonTensorHandleFactory.cpp.
Referenced by NeonTensorHandleFactory::CreateTensorHandle(), and NeonTensorHandleFactory::NeonTensorHandleFactory().
|
overridevirtual |
Implements ITensorHandleFactory.
Definition at line 95 of file NeonTensorHandleFactory.cpp.
Referenced by NeonTensorHandleFactory::NeonTensorHandleFactory().