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