1 files changed, 3 insertions, 0 deletions
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;