2 files changed, 16 insertions, 0 deletions
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp
index ef9f4e7522..22fc0a3ed4 100644
--- a/src/armnn/Network.cpp
+++ b/src/armnn/Network.cpp
@@ -2949,4 +2949,17 @@ OptimizedNetworkImpl::~OptimizedNetworkImpl()
 {
 }
 
+void IOptimizedNetwork::ExecuteStrategy(IStrategy &strategy) const
+{
+    pOptimizedNetworkImpl->ExecuteStrategy(strategy);
+}
+
+void OptimizedNetworkImpl::ExecuteStrategy(IStrategy &strategy) const
+{
+    for (auto layer : GetGraph())
+    {
+        layer->ExecuteStrategy(strategy);
+    };
+}
+
 } // namespace armnn
diff --git a/src/armnn/OptimizedNetworkImpl.hpp b/src/armnn/OptimizedNetworkImpl.hpp
index cb0dc4c8cf..45809d5619 100644
--- a/src/armnn/OptimizedNetworkImpl.hpp
+++ b/src/armnn/OptimizedNetworkImpl.hpp
@@ -25,8 +25,11 @@ public:
     virtual size_t GetNumOutputs() const;
 
     Graph& GetGraph() { return *m_Graph; }
+    Graph& GetGraph() const { return *m_Graph; }
     ModelOptions& GetModelOptions() { return m_ModelOptions; }
 
+    void ExecuteStrategy(IStrategy& strategy) const;
+
 private:
     std::unique_ptr<Graph> m_Graph;
     arm::pipe::ProfilingGuid m_Guid;