23.02
|
Go to the documentation of this file.
10 #include <unordered_map>
20 class IProfilingService;
28 class IBackendInternal;
29 class ICustomAllocator;
30 class IMemoryOptimizerStrategy;
49 std::unordered_map<BackendId, std::shared_ptr<ICustomAllocator>>
GetAllocators();
82 std::unordered_map<BackendId, std::shared_ptr<ICustomAllocator>> m_CustomMemoryAllocatorMap;
83 std::unordered_map<BackendId, std::shared_ptr<IMemoryOptimizerStrategy>> m_MemoryOptimizerStrategyMap;
static void Swap(BackendRegistry &instance, FactoryStorage &other)
For testing only.
virtual ~BackendRegistry()
BackendRegistry & BackendRegistryInstance()
std::unordered_map< BackendId, std::shared_ptr< IMemoryOptimizerStrategy > > MemoryOptimizerStrategiesMapRef
void RegisterAllocator(const BackendId &id, std::shared_ptr< ICustomAllocator > alloc)
FactoryFunction GetFactory(const BackendId &id) const
void DeregisterAllocator(const BackendId &id)
void SetProfilingService(armnn::Optional< arm::pipe::IProfilingService & > profilingService)
BackendIdSet GetBackendIds() const
Copyright (c) 2021 ARM Limited and Contributors.
void DeregisterMemoryOptimizerStrategy(const BackendId &id)
std::unique_ptr< IBackendInternal > IBackendInternalUniquePtr
StaticRegistryInitializer(BackendRegistry &instance, const BackendId &id, FactoryFunction factory)
void RegisterMemoryOptimizerStrategy(const BackendId &id, std::shared_ptr< IMemoryOptimizerStrategy > strategy)
std::unordered_map< BackendId, FactoryFunction > FactoryStorage
void Deregister(const BackendId &id)
std::string GetBackendIdsAsString() const
bool IsBackendRegistered(const BackendId &id) const
IBackendInternalUniquePtr PointerType
std::function< PointerType()> FactoryFunction
std::unordered_set< BackendId > BackendIdSet
MemoryOptimizerStrategiesMapRef GetMemoryOptimizerStrategies()
void Register(const BackendId &id, FactoryFunction factory)
std::unordered_map< BackendId, std::shared_ptr< ICustomAllocator > > GetAllocators()