aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/Graph.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/Graph.hpp')
-rw-r--r--src/armnn/Graph.hpp2
1 files changed, 2 insertions, 0 deletions
diff --git a/src/armnn/Graph.hpp b/src/armnn/Graph.hpp
index 8046977411..dd6a825f57 100644
--- a/src/armnn/Graph.hpp
+++ b/src/armnn/Graph.hpp
@@ -161,6 +161,8 @@ public:
/// and relinking them via an intermediary copy layers.
void AddCopyLayers();
+ /// 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.
void SubstituteSubGraph(std::unique_ptr<SubGraph> subGraph, IConnectableLayer* substituteLayer);
void SubstituteSubGraph(std::unique_ptr<SubGraph> subGraph, const SubGraph& substituteSubGraph);