diff options
Diffstat (limited to 'include/armnn/BackendRegistry.hpp')
-rw-r--r-- | include/armnn/BackendRegistry.hpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/include/armnn/BackendRegistry.hpp b/include/armnn/BackendRegistry.hpp index 1aaa11c55c..fe6451cde0 100644 --- a/include/armnn/BackendRegistry.hpp +++ b/include/armnn/BackendRegistry.hpp @@ -6,6 +6,7 @@ #include <armnn/Types.hpp> #include <armnn/BackendId.hpp> +#include <armnn/Optional.hpp> #include <memory> #include <unordered_map> @@ -14,6 +15,10 @@ namespace armnn { +namespace profiling +{ + class ProfilingService; +} class IBackendInternal; using IBackendInternalUniquePtr = std::unique_ptr<IBackendInternal>; @@ -29,6 +34,7 @@ public: size_t Size() const; BackendIdSet GetBackendIds() const; std::string GetBackendIdsAsString() const; + void SetProfilingService(armnn::Optional<profiling::ProfilingService&> profilingService); BackendRegistry() {} virtual ~BackendRegistry() {} @@ -56,6 +62,7 @@ private: BackendRegistry& operator=(const BackendRegistry&) = delete; FactoryStorage m_Factories; + armnn::Optional<profiling::ProfilingService&> m_ProfilingService; }; BackendRegistry& BackendRegistryInstance(); |