45 std::vector<std::unique_ptr<ITensorHandleFactory>> m_Factories;
46 std::vector<std::shared_ptr<IMemoryManager>> m_MemoryManagers;
void RegisterMemoryManager(std::shared_ptr< IMemoryManager > memoryManger)
Register a memory manager with shared ownership.
void RegisterFactory(std::unique_ptr< ITensorHandleFactory > allocator)
Register a TensorHandleFactory and transfer ownership.
void AquireMemory()
Aquire memory required for inference.
Copyright (c) 2020 ARM Limited.
void ReleaseMemory()
Release memory required for inference.
ITensorHandleFactory * GetFactory(ITensorHandleFactory::FactoryId id) const
Find a TensorHandleFactory by Id Returns nullptr if not found.
TensorHandleFactoryRegistry()=default