From adddddb6cbcb777d92a8c464c9ad0cb9aecc76a3 Mon Sep 17 00:00:00 2001 From: Matteo Martincigh Date: Thu, 24 Jan 2019 14:06:23 +0000 Subject: IVGCVSW-2458 Refactor the Optimize function (Network.cpp) so that subgraphs are optimized by the backends * Added a new method OptimizeSubGraph to the backend interface * Refactored the Optimize function so that the backend-specific optimization is performed by the backend itself (through the new OptimizeSubGraph interface method) * Added a new ApplyBackendOptimizations function to apply the new changes * Added some new convenient constructors to the SubGraph class * Added AddLayer method and a pointer to the parent graph to the SubGraph class * Updated the sub-graph unit tests to match the changes * Added SelectSubGraphs and ReplaceSubGraphConnections overloads that work with sub-graphs * Removed unused code and minor refactoring where necessary Change-Id: I46181794c6a9e3b10558944f804e06a8f693a6d0 --- src/backends/neon/NeonBackend.hpp | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'src/backends/neon/NeonBackend.hpp') diff --git a/src/backends/neon/NeonBackend.hpp b/src/backends/neon/NeonBackend.hpp index 3b1d186a06..634704571c 100644 --- a/src/backends/neon/NeonBackend.hpp +++ b/src/backends/neon/NeonBackend.hpp @@ -30,6 +30,9 @@ public: IBackendInternal::Optimizations GetOptimizations() const override; IBackendInternal::ILayerSupportSharedPtr GetLayerSupport() const override; + + IBackendInternal::SubGraphUniquePtr OptimizeSubGraph(const SubGraph& subGraph, + bool& optimizationAttempted) const override; }; -} // namespace armnn \ No newline at end of file +} // namespace armnn -- cgit v1.2.1