diff options
Diffstat (limited to 'src/armnn/SubGraphSelector.cpp')
-rw-r--r-- | src/armnn/SubGraphSelector.cpp | 2 |
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), |