aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/SubgraphView.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/SubgraphView.cpp')
-rw-r--r--src/armnn/SubgraphView.cpp18
1 files changed, 18 insertions, 0 deletions
diff --git a/src/armnn/SubgraphView.cpp b/src/armnn/SubgraphView.cpp
index 9426f1eefc..a87cc9b268 100644
--- a/src/armnn/SubgraphView.cpp
+++ b/src/armnn/SubgraphView.cpp
@@ -92,6 +92,17 @@ SubgraphView::SubgraphView(IConnectableLayer* layer)
CheckSubgraph();
}
+SubgraphView& SubgraphView::operator=(SubgraphView&& other)
+{
+ m_InputSlots = std::move(other.m_InputSlots);
+ m_OutputSlots = std::move(other.m_OutputSlots);
+ m_Layers = std::move(other.m_Layers);
+
+ CheckSubgraph();
+
+ return *this;
+}
+
void SubgraphView::CheckSubgraph()
{
// Check for invalid or duplicate input slots
@@ -179,4 +190,11 @@ SubgraphView::ConstIterator SubgraphView::cend() const
return end();
}
+void SubgraphView::Clear()
+{
+ m_InputSlots.clear();
+ m_OutputSlots.clear();
+ m_Layers.clear();
+}
+
} // namespace armnn