diff options
Diffstat (limited to 'include/armnn/INetwork.hpp')
-rw-r--r-- | include/armnn/INetwork.hpp | 3 |
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; |