diff options
| author | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-01-24 14:06:23 +0000 |
|---|---|---|
| committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-01-30 14:03:28 +0000 |
| commit | adddddb6cbcb777d92a8c464c9ad0cb9aecc76a3 (patch) | |
| tree | b15de32bf9f8612f66e1ae23d2f8009e80e7d0e6 /src/backends/cl | |
| parent | d089b74bebbcc8518fb0f4eacb7e6569ae170199 (diff) | |
| download | armnn-adddddb6cbcb777d92a8c464c9ad0cb9aecc76a3.tar.gz | |
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
Diffstat (limited to 'src/backends/cl')
| -rw-r--r-- | src/backends/cl/ClBackend.cpp | 9 | ||||
| -rw-r--r-- | src/backends/cl/ClBackend.hpp | 5 |
2 files changed, 13 insertions, 1 deletions
diff --git a/src/backends/cl/ClBackend.cpp b/src/backends/cl/ClBackend.cpp index 2f9dfa9755..dfa5e7c777 100644 --- a/src/backends/cl/ClBackend.cpp +++ b/src/backends/cl/ClBackend.cpp @@ -80,4 +80,13 @@ IBackendInternal::ILayerSupportSharedPtr ClBackend::GetLayerSupport() const return layerSupport; } +IBackendInternal::SubGraphUniquePtr ClBackend::OptimizeSubGraph(const SubGraph& subGraph, + bool& optimizationAttempted) const +{ + // Not trying to optimize the given sub-graph + optimizationAttempted = false; + + return SubGraphUniquePtr{}; +} + } // namespace armnn diff --git a/src/backends/cl/ClBackend.hpp b/src/backends/cl/ClBackend.hpp index 84b5b9a9d2..8133ce49f6 100644 --- a/src/backends/cl/ClBackend.hpp +++ b/src/backends/cl/ClBackend.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 |