[neurun] Enable setting the data type of mode operand::Object (#4881)

This commit Enable setting the data type of made operand::Object.

Signed-off-by: jiseob.jang <jiseob.jang@samsung.com>

diff --git a/runtimes/neurun/core/include/model/operand/Info.h b/runtimes/neurun/core/include/model/operand/Info.h
index 9065196..cda2ddf 100644 (file)
--- a/runtimes/neurun/core/include/model/operand/Info.h
+++ b/runtimes/neurun/core/include/model/operand/Info.h
@@ -43,6 +43,8 @@ public:
 public:
   const Shape &shape() const { return _shape; }
   const TypeInfo &typeInfo() const { return _typeInfo; }
+  void type(const DataType &type) { _typeInfo.type(type); }
+
   size_t total_size() const
   {
     const auto &dims = _shape.dims();

diff --git a/runtimes/neurun/core/include/model/operand/Object.h b/runtimes/neurun/core/include/model/operand/Object.h
index 558ebd9..4581ad8 100644 (file)
--- a/runtimes/neurun/core/include/model/operand/Object.h
+++ b/runtimes/neurun/core/include/model/operand/Object.h
@@ -26,6 +26,7 @@
 #include "Info.h"
 #include "graph/operand/ParentInfo.h" // TODO Remove this dependency
 #include "model/operation/IndexList.h"
+#include "model/operand/DataType.h"
 
 namespace neurun
 {
@@ -70,6 +71,9 @@ public:
   void removeDef(const operation::Index &idx);
 
 public:
+  void type(const DataType &type) { _info.type(type); };
+
+public:
   void data(std::unique_ptr<Data> &&data) { _data = std::move(data); }
   const Data &data(void) const
   {
@@ -117,7 +121,7 @@ public:
   graph::operand::ParentInfo *parent_info();
 
 private:
-  const Info _info;
+  Info _info;
   std::unique_ptr<Data> _data;
   Usage _usage;
 

diff --git a/runtimes/neurun/core/include/model/operand/TypeInfo.h b/runtimes/neurun/core/include/model/operand/TypeInfo.h
index 5cbe213..10a47d9 100644 (file)
--- a/runtimes/neurun/core/include/model/operand/TypeInfo.h
+++ b/runtimes/neurun/core/include/model/operand/TypeInfo.h
@@ -43,6 +43,9 @@ public:
   float scale() const { return _scale; }
   int32_t offset() const { return _offset; }
 
+public:
+  void type(const DataType &type) { _type = type; }
+
 private:
   DataType _type;
   float _scale;