45 std::vector<std::unique_ptr<ITensorHandleFactory>> m_Factories;
46 std::vector<std::shared_ptr<IMemoryManager>> m_MemoryManagers;
void RegisterFactory(std::unique_ptr< ITensorHandleFactory > allocator)
Register a TensorHandleFactory and transfer ownership.
void ReleaseMemory()
Release memory required for inference.
ITensorHandleFactory * GetFactory(ITensorHandleFactory::FactoryId id) const
void AquireMemory()
Aquire memory required for inference.
TensorHandleFactoryRegistry()=default
void RegisterMemoryManager(std::shared_ptr< IMemoryManager > memoryManger)
Register a memory manager with shared ownership.