diff options
Diffstat (limited to 'src/armnn/SubGraph.hpp')
-rw-r--r-- | src/armnn/SubGraph.hpp | 13 |
1 files changed, 11 insertions, 2 deletions
diff --git a/src/armnn/SubGraph.hpp b/src/armnn/SubGraph.hpp index 312bb115eb..d22377daff 100644 --- a/src/armnn/SubGraph.hpp +++ b/src/armnn/SubGraph.hpp @@ -8,7 +8,7 @@ #include "Layer.hpp" #include <vector> -#include <unordered_set> +#include <list> namespace armnn { @@ -24,7 +24,7 @@ class SubGraph final public: using InputSlots = std::vector<InputSlot *>; using OutputSlots = std::vector<OutputSlot *>; - using Layers = std::unordered_set<Layer *>; + using Layers = std::list<Layer*>; SubGraph(); SubGraph(InputSlots && inputs, @@ -44,6 +44,15 @@ public: unsigned int GetNumInputSlots() const; unsigned int GetNumOutputSlots() const; + Layers::iterator begin(); + Layers::iterator end(); + + Layers::const_iterator begin() const; + Layers::const_iterator end() const; + + Layers::const_iterator cbegin() const; + Layers::const_iterator cend() const; + private: InputSlots m_InputSlots; OutputSlots m_OutputSlots; |