aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/SubGraphSelector.cpp
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/SubGraphSelector.cpp
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/SubGraphSelector.cpp')
-rw-r--r--src/armnn/SubGraphSelector.cpp2
1 files changed, 2 insertions, 0 deletions
diff --git a/src/armnn/SubGraphSelector.cpp b/src/armnn/SubGraphSelector.cpp
index 4abf01c88f..4b3f79921f 100644
--- a/src/armnn/SubGraphSelector.cpp
+++ b/src/armnn/SubGraphSelector.cpp
@@ -174,6 +174,8 @@ SubGraphSelector::SelectSubGraphs(SubGraph& subGraph, const LayerSelectorFunctio
infoPtr->CollectNonSelectedOutputSlots(outputs, selector);
layers.push_back(infoPtr->m_Layer);
}
+ // Create a new sub-graph with the new lists of input/output slots and layer, using
+ // the given sub-graph as a reference of which parent graph to use
result.emplace_back(std::make_unique<SubGraph>(subGraph,
std::move(inputs),
std::move(outputs),