ArmNN
NotReleased
|
#include <TensorHandleFactoryRegistry.hpp>
Public Member Functions | |
TensorHandleFactoryRegistry ()=default | |
TensorHandleFactoryRegistry (const TensorHandleFactoryRegistry &other)=delete | |
TensorHandleFactoryRegistry (TensorHandleFactoryRegistry &&other)=delete | |
void | RegisterFactory (std::unique_ptr< ITensorHandleFactory > allocator) |
Register a TensorHandleFactory and transfer ownership. More... | |
void | RegisterMemoryManager (std::shared_ptr< IMemoryManager > memoryManger) |
Register a memory manager with shared ownership. More... | |
ITensorHandleFactory * | GetFactory (ITensorHandleFactory::FactoryId id) const |
void | AquireMemory () |
Aquire memory required for inference. More... | |
void | ReleaseMemory () |
Release memory required for inference. More... | |
Definition at line 20 of file TensorHandleFactoryRegistry.hpp.
|
default |
|
delete |
|
delete |
void AquireMemory | ( | ) |
Aquire memory required for inference.
Definition at line 52 of file TensorHandleFactoryRegistry.cpp.
ITensorHandleFactory * GetFactory | ( | ITensorHandleFactory::FactoryId | id | ) | const |
Find a TensorHandleFactory by Id Returns nullptr if not found
Definition at line 39 of file TensorHandleFactoryRegistry.cpp.
Referenced by Graph::AddCompatibilityLayers(), armnn::CalculateEdgeStrategy(), armnn::CalculateSlotOption(), armnn::CalculateSlotOptionForInput(), ConcatLayer::CreateTensorHandles(), SplitterLayer::CreateTensorHandles(), Layer::CreateTensorHandles(), and armnn::RequiresCopy().
void RegisterFactory | ( | std::unique_ptr< ITensorHandleFactory > | allocator | ) |
Register a TensorHandleFactory and transfer ownership.
Definition at line 12 of file TensorHandleFactoryRegistry.cpp.
Referenced by ClBackend::RegisterTensorHandleFactories(), NeonBackend::RegisterTensorHandleFactories(), and RefBackend::RegisterTensorHandleFactories().
void RegisterMemoryManager | ( | std::shared_ptr< IMemoryManager > | memoryManger | ) |
Register a memory manager with shared ownership.
Definition at line 34 of file TensorHandleFactoryRegistry.cpp.
Referenced by ClBackend::CreateWorkloadFactory(), RefBackend::CreateWorkloadFactory(), NeonBackend::CreateWorkloadFactory(), ClBackend::RegisterTensorHandleFactories(), NeonBackend::RegisterTensorHandleFactories(), and RefBackend::RegisterTensorHandleFactories().
void ReleaseMemory | ( | ) |
Release memory required for inference.
Definition at line 60 of file TensorHandleFactoryRegistry.cpp.