From e155bbf0e9be6b4d7974297585a59207cd89b00a Mon Sep 17 00:00:00 2001
From: Derek Lamberti <derek.lamberti@arm.com>
Date: Wed, 13 Oct 2021 14:32:12 +0100
Subject: 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
---
 include/armnn/INetwork.hpp | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'include/armnn/INetwork.hpp')

diff --git a/include/armnn/INetwork.hpp b/include/armnn/INetwork.hpp
index 5027818623..ab92f05112 100644
--- a/include/armnn/INetwork.hpp
+++ b/include/armnn/INetwork.hpp
@@ -715,6 +715,7 @@ class WorkingMemHandle;
 struct BackendSettings;
 struct OptimizationResult;
 class OptimizedNetworkImpl;
+class IProfiler;
 class IOptimizedNetwork
 {
 public:
@@ -732,6 +733,8 @@ public:
     IOptimizedNetwork(std::unique_ptr<OptimizedNetworkImpl> impl);
     ~IOptimizedNetwork();
 
+    const std::shared_ptr<IProfiler>& GetProfiler() const;
+
 protected:
     friend class LoadedNetwork;
 
-- 
cgit v1.2.1