diff options
Diffstat (limited to 'src/armnn/Graph.hpp')
| -rw-r--r-- | src/armnn/Graph.hpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/armnn/Graph.hpp b/src/armnn/Graph.hpp index 47e0e3b317..f8113375c9 100644 --- a/src/armnn/Graph.hpp +++ b/src/armnn/Graph.hpp @@ -191,7 +191,8 @@ public: /// Modifies the graph in-place, removing edges connecting layers using different compute devices, /// and relinking them via an intermediary copy layers. - void AddCopyLayers(); + void AddCopyLayers(std::map<BackendId, std::unique_ptr<class IBackendInternal>>& backends, + TensorHandleFactoryRegistry& registry); /// Substitutes the given sub-graph with either a new layer or a new sub-graph. /// In either case, the given layer or all the layers in the given sub-graph must belong to this graph. |