diff options
Diffstat (limited to 'src/backends/BackendRegistry.hpp')
-rw-r--r-- | src/backends/BackendRegistry.hpp | 13 |
1 files changed, 7 insertions, 6 deletions
diff --git a/src/backends/BackendRegistry.hpp b/src/backends/BackendRegistry.hpp index ff01d21715..e2c526d293 100644 --- a/src/backends/BackendRegistry.hpp +++ b/src/backends/BackendRegistry.hpp @@ -5,7 +5,6 @@ #pragma once #include <armnn/Types.hpp> -#include <string> #include <functional> #include <memory> #include <unordered_map> @@ -21,21 +20,23 @@ public: using FactoryFunction = std::function<IBackendUniquePtr()>; static BackendRegistry& Instance(); - void Register(const std::string& name, FactoryFunction factory); - FactoryFunction GetFactory(const std::string& name) const; + + void Register(const BackendId& id, FactoryFunction factory); + FactoryFunction GetFactory(const BackendId& id) const; struct Helper { - Helper(const std::string& name, FactoryFunction factory) + Helper(const BackendId& id, FactoryFunction factory) { - BackendRegistry::Instance().Register(name, factory); + BackendRegistry::Instance().Register(id, factory); } }; size_t Size() const { return m_BackendFactories.size(); } + BackendIdSet GetBackendIds() const; protected: - using FactoryStorage = std::unordered_map<std::string, FactoryFunction>; + using FactoryStorage = std::unordered_map<BackendId, FactoryFunction>; // For testing only static void Swap(FactoryStorage& other); |