[neurun] Introduce iterate method for Context (#3425)

Introduce `iterate` method for `codegen::operand::Context`.

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

diff --git a/runtimes/neurun/src/codegen/operand/Context.cc b/runtimes/neurun/src/codegen/operand/Context.cc
index 76ba59f..17acca0 100644 (file)
--- a/runtimes/neurun/src/codegen/operand/Context.cc
+++ b/runtimes/neurun/src/codegen/operand/Context.cc
@@ -34,6 +34,14 @@ Context &Context::set(const graph::operand::Index &id,
   return (*this);
 }
 
+void Context::iterate(const std::function<void(int, backend::operand::IObject &)> &fn)
+{
+  for (auto &e : _objects)
+  {
+    fn(e.first, *e.second);
+  }
+}
+
 } // namespace operand
 } // namespace codegen
 } // namespace neurun

diff --git a/runtimes/neurun/src/codegen/operand/Context.h b/runtimes/neurun/src/codegen/operand/Context.h
index b1000a8..3437da2 100644 (file)
--- a/runtimes/neurun/src/codegen/operand/Context.h
+++ b/runtimes/neurun/src/codegen/operand/Context.h
@@ -52,6 +52,8 @@ public:
     return _objects.at(ind.asInt());
   }
 
+  void iterate(const std::function<void(int, backend::operand::IObject &)> &fn);
+
 private:
   std::map<int, std::shared_ptr<backend::operand::IObject>> _objects;
 };