diff options
Diffstat (limited to 'src/armnn/Network.hpp')
| -rw-r--r-- | src/armnn/Network.hpp | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/src/armnn/Network.hpp b/src/armnn/Network.hpp index ee84d7b33b..a2ef0d8364 100644 --- a/src/armnn/Network.hpp +++ b/src/armnn/Network.hpp @@ -20,6 +20,7 @@ #include "Graph.hpp" #include "Layer.hpp" #include "OptimizedNetworkImpl.hpp" +#include "SubgraphView.hpp" namespace armnn { @@ -301,4 +302,11 @@ OptimizationResult AssignBackends(OptimizedNetworkImpl* optNetObjPtr, Graph::Iterator& lastLayer, Optional<std::vector<std::string>&> errMessages); + +OptimizationResult AssignBackends(OptimizedNetworkImpl* optNetObjPtr, + BackendSettings& backendSettings, + SubgraphView::IConnectableLayerIterator& firstLayer, + SubgraphView::IConnectableLayerIterator& lastLayer, + Optional<std::vector<std::string>&> errMessages); + } // namespace armnn |