diff options
Diffstat (limited to 'src/armnn')
-rw-r--r-- | src/armnn/Network.cpp | 13 | ||||
-rw-r--r-- | src/armnn/OptimizedNetworkImpl.hpp | 3 |
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; |