aboutsummaryrefslogtreecommitdiff
path: root/include/armnn/BackendRegistry.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'include/armnn/BackendRegistry.hpp')
-rw-r--r--include/armnn/BackendRegistry.hpp18
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;
};