diff options
author | Derek Lamberti <derek.lamberti@arm.com> | 2021-10-13 14:32:12 +0100 |
---|---|---|
committer | Derek Lamberti <derek.lamberti@arm.com> | 2021-10-15 15:29:43 +0100 |
commit | e155bbf0e9be6b4d7974297585a59207cd89b00a (patch) | |
tree | 78f5d5f4f2a3a8551854fdd46eac7f599e371bd1 /src/armnn/LoadedNetwork.cpp | |
parent | 827e4bf519ec1e4af2fb0b32d476b8b1c73b877c (diff) | |
download | armnn-e155bbf0e9be6b4d7974297585a59207cd89b00a.tar.gz |
Refactor: Profiler moved to Graph
* This is to enable later work to instrument the Optimizer.
Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>
Change-Id: I2cf1fe022e0d100d6d8705adfbb8cab3ffc96a86
Diffstat (limited to 'src/armnn/LoadedNetwork.cpp')
-rw-r--r-- | src/armnn/LoadedNetwork.cpp | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/src/armnn/LoadedNetwork.cpp b/src/armnn/LoadedNetwork.cpp index c161ed35d5..4688b6eea4 100644 --- a/src/armnn/LoadedNetwork.cpp +++ b/src/armnn/LoadedNetwork.cpp @@ -122,13 +122,13 @@ LoadedNetwork::LoadedNetwork(std::unique_ptr<IOptimizedNetwork> net, m_TensorHandleFactoryRegistry(), m_ProfilingService(profilingService) { - // Create a profiler and register it for the current thread. - m_Profiler = std::make_shared<IProfiler>(); - ProfilerManager::GetInstance().RegisterProfiler(m_Profiler.get()); + // Get the profiler and register it for the current thread. + const std::shared_ptr<IProfiler>& profiler = m_OptimizedNetwork->GetProfiler(); + ProfilerManager::GetInstance().RegisterProfiler(profiler.get()); - m_Profiler->EnableProfiling(networkProperties.m_ProfilingEnabled); + profiler->EnableProfiling(networkProperties.m_ProfilingEnabled); - m_Profiler->EnableNetworkDetailsToStdOut(networkProperties.m_OutputNetworkDetailsMethod); + profiler->EnableNetworkDetailsToStdOut(networkProperties.m_OutputNetworkDetailsMethod); Graph& order = m_OptimizedNetwork->pOptimizedNetworkImpl->GetGraph().TopologicalSort(); //First create tensor handlers, backends and workload factories. |