22 for (
auto& registeredFactory : m_Factories)
24 if (
id == registeredFactory->GetId())
31 m_Factories.push_back(std::move(newFactory));
36 m_MemoryManagers.push_back(memoryManger);
41 for (
auto& factory : m_Factories)
43 if (factory->GetId() == id)
55 for (
auto& factory : m_Factories)
57 if (factory->GetId() ==
id && factory->GetImportFlags() ==
static_cast<MemorySourceFlags>(memSource))
68 for (
auto& mgr : m_MemoryManagers)
76 for (
auto& mgr : 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.
unsigned int MemorySourceFlags
Copyright (c) 2021 ARM Limited and Contributors.
void ReleaseMemory()
Release memory required for inference.
MemorySource
Define the Memory Source to reduce copies.
ITensorHandleFactory * GetFactory(ITensorHandleFactory::FactoryId id) const
Find a TensorHandleFactory by Id Returns nullptr if not found.