diff options
Diffstat (limited to 'include/armnn/BackendRegistry.hpp')
-rw-r--r-- | include/armnn/BackendRegistry.hpp | 18 |
1 files changed, 12 insertions, 6 deletions
diff --git a/include/armnn/BackendRegistry.hpp b/include/armnn/BackendRegistry.hpp index 0d09607de2..08e164c549 100644 --- a/include/armnn/BackendRegistry.hpp +++ b/include/armnn/BackendRegistry.hpp @@ -15,13 +15,19 @@ #include <unordered_map> #include <functional> -namespace armnn +namespace arm +{ +namespace pipe { -namespace profiling +class ProfilingService; + +} // namespace arm +} // namespace pipe + +namespace armnn { - class ProfilingService; -} + class IBackendInternal; using IBackendInternalUniquePtr = std::unique_ptr<IBackendInternal>; using MemoryOptimizerStrategiesMapRef = std::unordered_map<BackendId, std::shared_ptr<IMemoryOptimizerStrategy>>; @@ -38,7 +44,7 @@ public: size_t Size() const; BackendIdSet GetBackendIds() const; std::string GetBackendIdsAsString() const; - void SetProfilingService(armnn::Optional<profiling::ProfilingService&> profilingService); + void SetProfilingService(armnn::Optional<arm::pipe::ProfilingService&> profilingService); void RegisterAllocator(const BackendId& id, std::shared_ptr<ICustomAllocator> alloc); std::unordered_map<BackendId, std::shared_ptr<ICustomAllocator>> GetAllocators(); void RegisterMemoryOptimizerStrategy(const BackendId& id, std::shared_ptr<IMemoryOptimizerStrategy> strategy); @@ -72,7 +78,7 @@ private: BackendRegistry& operator=(const BackendRegistry&) = delete; FactoryStorage m_Factories; - armnn::Optional<profiling::ProfilingService&> m_ProfilingService; + armnn::Optional<arm::pipe::ProfilingService&> m_ProfilingService; std::unordered_map<BackendId, std::shared_ptr<ICustomAllocator>> m_CustomMemoryAllocatorMap; std::unordered_map<BackendId, std::shared_ptr<IMemoryOptimizerStrategy>> m_MemoryOptimizerStrategyMap; }; |