[neurun] Support iteration for operand::Object (#2608)

Support const/non-const iteration for operand::Object.

Signed-off-by: Hanjoung Lee <hanjoung.lee@samsung.com>

diff --git a/runtimes/neurun/src/graph/operand/Set.cc b/runtimes/neurun/src/graph/operand/Set.cc
index 862e011..4f8263c 100644 (file)
--- a/runtimes/neurun/src/graph/operand/Set.cc
+++ b/runtimes/neurun/src/graph/operand/Set.cc
@@ -31,6 +31,22 @@ Object &Set::at(const Index &index) { return *(_objects.at(index)); }
 
 bool Set::exist(const Index &index) const { return index.value() < _objects.size(); }
 
+void Set::iterate(const std::function<void(const Index &, const Object &)> &fn) const
+{
+  for (const auto &e : _objects)
+  {
+    fn(e.first, *e.second);
+  }
+}
+
+void Set::iterate(const std::function<void(const Index &, Object &)> &fn)
+{
+  for (auto &e : _objects)
+  {
+    fn(e.first, *e.second);
+  }
+}
+
 } // namespace operand
 } // namespace graph
 } // namespace neurun

diff --git a/runtimes/neurun/src/graph/operand/Set.h b/runtimes/neurun/src/graph/operand/Set.h
index 754aec1..5b310cd 100644 (file)
--- a/runtimes/neurun/src/graph/operand/Set.h
+++ b/runtimes/neurun/src/graph/operand/Set.h
@@ -26,6 +26,8 @@ public:
   const Object &at(const Index &) const;
   Object &at(const Index &);
   bool exist(const Index &) const;
+  void iterate(const std::function<void(const Index &, const Object &)> &fn) const;
+  void iterate(const std::function<void(const Index &, Object &)> &fn);
 
 private:
   const Index generateIndex();