From: 박종현/동작제어Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com>
Date: Wed, 29 Aug 2018 04:30:11 +0000 (+0900)
Subject: [enco] Record ANN input/output opreands (#1227)
X-Git-Tag: nncc_backup~2024
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=b6df9490f205ba4f3ea6457def19f67f67c09b78;p=platform%2Fcore%2Fml%2Fnnfw.git

[enco] Record ANN input/output opreands (#1227)

This commit extends ann::Module class with input and output methods
which records the input and output operands of android NN model.

Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>
---

diff --git a/contrib/enco/core/src/ANN/IR/InputList.h b/contrib/enco/core/src/ANN/IR/InputList.h
new file mode 100644
index 0000000..23de60a
--- /dev/null
+++ b/contrib/enco/core/src/ANN/IR/InputList.h
@@ -0,0 +1,15 @@
+#ifndef __ANN_IR_INPUT_LIST_H__
+#define __ANN_IR_INPUT_LIST_H__
+
+#include "ANN/IR/OperandID.h"
+
+#include <vector>
+
+namespace ann
+{
+
+using InputList = std::vector<OperandID>;
+
+} // namespace ann
+
+#endif // __ANN_IR_INPUT_LIST_H__
diff --git a/contrib/enco/core/src/ANN/IR/Module.h b/contrib/enco/core/src/ANN/IR/Module.h
index 181dad6..12472cb 100644
--- a/contrib/enco/core/src/ANN/IR/Module.h
+++ b/contrib/enco/core/src/ANN/IR/Module.h
@@ -4,6 +4,8 @@
 #include "ANN/IR/WeightInventory.h"
 #include "ANN/IR/OperandInventory.h"
 #include "ANN/IR/OperationInventory.h"
+#include "ANN/IR/InputList.h"
+#include "ANN/IR/OutputList.h"
 
 namespace ann
 {
@@ -23,10 +25,18 @@ public:
   OperationInventory *operation(void) { return &_operation; }
   const OperationInventory *operation(void) const { return &_operation; }
 
+  InputList *input(void) { return &_input; }
+  const InputList *input(void) const { return &_input; }
+
+  OutputList *output(void) { return &_output; }
+  const OutputList *output(void) const { return &_output; }
+
 private:
   WeightInventory _weight;
   OperandInventory _operand;
   OperationInventory _operation;
+  InputList _input;
+  OutputList _output;
 };
 
 } // namespace ann
diff --git a/contrib/enco/core/src/ANN/IR/OutputList.h b/contrib/enco/core/src/ANN/IR/OutputList.h
new file mode 100644
index 0000000..a8881d0
--- /dev/null
+++ b/contrib/enco/core/src/ANN/IR/OutputList.h
@@ -0,0 +1,15 @@
+#ifndef __ANN_IR_OUTPUT_LIST_H__
+#define __ANN_IR_OUTPUT_LIST_H__
+
+#include "ANN/IR/OperandID.h"
+
+#include <vector>
+
+namespace ann
+{
+
+using OutputList = std::vector<OperandID>;
+
+} // namespace ann
+
+#endif // __ANN_IR_OUTPUT_LIST_H__