[neurun] Graph : Introduce Operation `Node` (#2206)
author이한종/동작제어Lab(SR)/Engineer/삼성전자 <hanjoung.lee@samsung.com>
Wed, 8 Aug 2018 08:24:15 +0000 (17:24 +0900)
committer박세희/동작제어Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Wed, 8 Aug 2018 08:24:15 +0000 (17:24 +0900)
Class `neurun::graph::operation::Node` is a wrapper class of
`internal::tflite::op::Node` with input/output tensor info.
Input/Output info will be used for graph traversing.

Signed-off-by: Hanjoung Lee <hanjoung.lee@samsung.com>
runtimes/neurun/src/graph/operation/Node.h [new file with mode: 0644]

diff --git a/runtimes/neurun/src/graph/operation/Node.h b/runtimes/neurun/src/graph/operation/Node.h
new file mode 100644 (file)
index 0000000..586516c
--- /dev/null
@@ -0,0 +1,29 @@
+#ifndef __NEURUN_GRAPH_OPERATION_NODE_H__
+#define __NEURUN_GRAPH_OPERATION_NODE_H__
+
+#include "graph/operand/IndexSet.h"
+#include "internal/op/Node.h"
+
+namespace neurun
+{
+namespace graph
+{
+namespace operation
+{
+
+struct Node
+{
+public:
+  virtual ~Node() = default;
+
+public:
+  virtual operand::IndexSet inputs() const = 0;
+  virtual operand::IndexSet outputs() const = 0;
+  virtual const ::internal::tflite::op::Node *op() const = 0;
+};
+
+} // namespace operation
+} // namespace graph
+} // namespace neurun
+
+#endif // __NEURUN_GRAPH_OPERATION_NODE_H__