aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/Graph.hpp
diff options
context:
space:
mode:
authorMatteo Martincigh <matteo.martincigh@arm.com>2019-01-31 12:09:49 +0000
committerMatteo Martincigh <matteo.martincigh@arm.com>2019-01-31 17:25:45 +0000
commit0c051f9b6b8d7d1602e81d2977dda449b6392642 (patch)
tree0d9fbca37c82bd2bc8d5799c3708bc70392455d6 /src/armnn/Graph.hpp
parent4a55201f9d24d6bed46781ec56ba8862ddc7c26b (diff)
downloadarmnn-0c051f9b6b8d7d1602e81d2977dda449b6392642.tar.gz
IVGCVSW-2471 Code cleanup where necessary
* Added comments for the new API functions * Added comments to some critical parts of the code * Removed unnecessary code Change-Id: I04d4e7373b45b8a3806cacc93a8bd09351c0972a Signed-off-by: Matteo Martincigh <matteo.martincigh@arm.com>
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);