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 --- include/armnn/BackendId.hpp | 2 ++ 1 file changed, 2 insertions(+) (limited to 'include/armnn') diff --git a/include/armnn/BackendId.hpp b/include/armnn/BackendId.hpp index 129cbb5d46..89725c165a 100644 --- a/include/armnn/BackendId.hpp +++ b/include/armnn/BackendId.hpp @@ -125,6 +125,8 @@ public: return m_Id < other.m_Id; } + bool IsCpuRef() const { return m_Id == GetComputeDeviceAsCString(Compute::CpuRef); } + const std::string& Get() const { return m_Id; } private: -- cgit v1.2.1