1 files changed, 8 insertions, 2 deletions
diff --git a/runtimes/neurun/src/frontend/wrapper/model.h b/runtimes/neurun/src/frontend/wrapper/model.h
index 3c7b027dc..2386a648d 100644
--- a/runtimes/neurun/src/frontend/wrapper/model.h
+++ b/runtimes/neurun/src/frontend/wrapper/model.h
@@ -30,12 +30,18 @@ public:
   neurun::graph::Graph &deref(void) { return *_model; }
   ResultCode finish();
   bool isFinished() { return !_model->isBuildingPhase(); }
-
-public:
   void release(std::shared_ptr<neurun::graph::Graph> &model) { model = _model; }
+  void setOptionalOperand(const neurun::model::operand::Index idx)
+  {
+    _optional_operands.insert(idx);
+  }
+
+private:
+  void fillOptionalOperand(void);
 
 private:
   std::shared_ptr<neurun::graph::Graph> _model;
+  std::unordered_set<neurun::model::operand::Index> _optional_operands;
 };
 
 #endif // __MODEL_H__