aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/SubgraphView.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/SubgraphView.hpp')
-rw-r--r--src/armnn/SubgraphView.hpp7
1 files changed, 6 insertions, 1 deletions
diff --git a/src/armnn/SubgraphView.hpp b/src/armnn/SubgraphView.hpp
index d4d92bbf6c..d86f1c1c93 100644
--- a/src/armnn/SubgraphView.hpp
+++ b/src/armnn/SubgraphView.hpp
@@ -31,7 +31,7 @@ public:
using ConstIterator = Layers::const_iterator;
/// Constructs a sub-graph from the entire given graph.
- SubgraphView(Graph& graph);
+ explicit SubgraphView(Graph& graph);
/// Constructs a sub-graph with the given arguments.
SubgraphView(InputSlots&& inputs, OutputSlots&& outputs, Layers&& layers);
@@ -45,6 +45,9 @@ public:
/// Constructs a sub-graph with only the given layer.
SubgraphView(IConnectableLayer* layer);
+ /// Move-assignment operator.
+ SubgraphView& operator=(SubgraphView&& other);
+
const InputSlots& GetInputSlots() const;
const OutputSlots& GetOutputSlots() const;
const Layers& GetLayers() const;
@@ -67,6 +70,8 @@ public:
ConstIterator cbegin() const;
ConstIterator cend() const;
+ void Clear();
+
private:
void CheckSubgraph();