diff options
Diffstat (limited to 'src/armnn/Runtime.hpp')
-rw-r--r-- | src/armnn/Runtime.hpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/src/armnn/Runtime.hpp b/src/armnn/Runtime.hpp index 2ad3c9633c..477b1169b1 100644 --- a/src/armnn/Runtime.hpp +++ b/src/armnn/Runtime.hpp @@ -14,6 +14,8 @@ #include <armnn/backends/DynamicBackend.hpp> +#include <ProfilingService.hpp> + #include <mutex> #include <unordered_map> @@ -80,6 +82,8 @@ public: private: friend void RuntimeLoadedNetworksReserve(armnn::Runtime* runtime); // See RuntimeTests.cpp + friend profiling::ProfilingService& GetProfilingService(armnn::Runtime* runtime); // See RuntimeTests.cpp + int GenerateNetworkId(); LoadedNetwork* GetLoadedNetworkPtr(NetworkId networkId) const; @@ -109,6 +113,9 @@ private: /// List of dynamic backends loaded in the runtime std::vector<DynamicBackendPtr> m_DynamicBackends; + + /// Profiling Service Instance + profiling::ProfilingService m_ProfilingService; }; } // namespace armnn |