From 61a5181e31a073a2b23cc76028fc24119d4970c7 Mon Sep 17 00:00:00 2001
From: Yu Yang <yuyang18@baidu.com>
Date: Mon, 9 Oct 2017 11:05:25 -0700
Subject: [PATCH] Add skeleton of Operator

---
 python/paddle/v2/framework/graph.py | 44 +++++++++++++++++++++++++++--
 1 file changed, 42 insertions(+), 2 deletions(-)

diff --git a/python/paddle/v2/framework/graph.py b/python/paddle/v2/framework/graph.py
index 5b93115b3e9..6f2a76a9835 100644
--- a/python/paddle/v2/framework/graph.py
+++ b/python/paddle/v2/framework/graph.py
@@ -1,6 +1,7 @@
 import paddle.v2.framework.core as core
+import collections
 
-__all__ = ['Block', 'Variable', 'Program']
+__all__ = ['Block', 'Variable', 'Program', 'Operator']
 
 
 class Variable(object):
@@ -24,6 +25,7 @@ class Variable(object):
             self.proto.set_lod_level(lod_level)
 
         self.block.vars[name] = self
+        self.op = None
 
     # TODO(yuyang18): Get methods
 
@@ -33,11 +35,37 @@ class Variable(object):
         return "_generated_var_%d" % uid
 
 
+class Operator(object):
+    def __init__(self,
+                 block,
+                 proto,
+                 type=None,
+                 inputs=None,
+                 outputs=None,
+                 attrs=None):
+        self.block = block
+        self.proto = proto
+        if type is not None:
+            # TODO.
+            pass
+        if inputs is not None:
+            # TODO
+            pass
+        if outputs is not None:
+            # TODO
+            pass
+        if attrs is not None:
+            # TODO
+            pass
+
+        # TODO: Getters
+
+
 class Block(object):
     def __init__(self, program, idx):
         self.proto = program.proto.block(idx)
         self.vars = dict()  # var_name --> var
-        self.ops = list()  # operator list
+        self.ops = collections.deque()  # operator list
         self.program = program
 
     @property
@@ -51,6 +79,18 @@ class Block(object):
     def create_var(self, *args, **kwargs):
         return Variable(self, *args, **kwargs)
 
+    def append_op(self, *args, **kwargs):
+        op_proto = self.proto.append_op()
+        op = Operator(self, op_proto, *args, **kwargs)
+        self.ops.append(op)
+        return op
+
+    def prepend_op(self, *args, **kwargs):
+        op_proto = self.proto.prepend_op()
+        op = Operator(self, op_proto, *args, **kwargs)
+        self.ops.appendleft(op)
+        return op
+
 
 class Program(object):
     @classmethod
-- 
GitLab