From 3184c907b2420e6c66485529f336251b2b62aecf Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Wed, 18 Mar 2020 10:57:30 +0000 Subject: IVGCVSW-4463 Change ProfilingService to a member of runtime from a singleton Signed-off-by: Sadik Armagan Change-Id: I345c39a10a4693a500aa1687d9a5cee76da791c3 --- src/armnn/LoadedNetwork.hpp | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) (limited to 'src/armnn/LoadedNetwork.hpp') diff --git a/src/armnn/LoadedNetwork.hpp b/src/armnn/LoadedNetwork.hpp index ab2c8be1cd..01e3442508 100644 --- a/src/armnn/LoadedNetwork.hpp +++ b/src/armnn/LoadedNetwork.hpp @@ -15,6 +15,7 @@ #include #include #include +#include #include #include @@ -43,7 +44,8 @@ public: static std::unique_ptr MakeLoadedNetwork(std::unique_ptr net, std::string & errorMessage, - const INetworkProperties& networkProperties); + const INetworkProperties& networkProperties, + profiling::ProfilingService& profilingService); // NOTE we return by reference as the purpose of this method is only to provide // access to the private m_Profiler and in theory we should not need to increment @@ -57,7 +59,9 @@ public: private: void AllocateWorkingMemory(); - LoadedNetwork(std::unique_ptr net, const INetworkProperties& networkProperties); + LoadedNetwork(std::unique_ptr net, + const INetworkProperties& networkProperties, + profiling::ProfilingService& profilingService); void EnqueueInput(const BindableLayer& layer, ITensorHandle* tensorHandle, const TensorInfo& tensorInfo); @@ -92,6 +96,8 @@ private: bool m_IsExportEnabled=false; TensorHandleFactoryRegistry m_TensorHandleFactoryRegistry; + + profiling::ProfilingService& m_ProfilingService; }; } -- cgit v1.2.1