提交 cffca923 编写于 作者: F fengjiayi

Change Proto to Desc

上级 5fddd288
...@@ -46,18 +46,18 @@ class Variable(object): ...@@ -46,18 +46,18 @@ class Variable(object):
if name is None: if name is None:
name = Variable._unique_var_name_() name = Variable._unique_var_name_()
self.proto = self.block.proto.new_var(name) self.desc = self.block.desc.new_var(name)
if shape is not None: if shape is not None:
self.proto.set_shape(shape) self.desc.set_shape(shape)
if dtype is not None: if dtype is not None:
# TODO(yuyang18): Convert dtype from numpy.dtype # TODO(yuyang18): Convert dtype from numpy.dtype
self.proto.set_data_type(dtype) self.desc.set_data_type(dtype)
if lod_level is not None: if lod_level is not None:
# TODO(yuyang18): set_lod_level is not defined. # TODO(yuyang18): set_lod_level is not defined.
self.proto.set_lod_level(lod_level) self.desc.set_lod_level(lod_level)
self.block.vars[name] = self self.block.vars[name] = self
self.op = None self.op = None
...@@ -121,31 +121,31 @@ class Operator(object): ...@@ -121,31 +121,31 @@ class Operator(object):
class Block(object): class Block(object):
def __init__(self, program, idx): def __init__(self, program, idx):
self.proto = program.proto.block(idx) self.desc = program.desc.block(idx)
self.vars = dict() # var_name --> var self.vars = dict() # var_name --> var
self.ops = collections.deque() # operator list self.ops = collections.deque() # operator list
self.program = program self.program = program
@property @property
def parent_idx(self): def parent_idx(self):
return self.proto.parent return self.desc.parent
@property @property
def idx(self): def idx(self):
return self.proto.id return self.desc.id
def create_var(self, *args, **kwargs): def create_var(self, *args, **kwargs):
return Variable(self, *args, **kwargs) return Variable(self, *args, **kwargs)
def append_op(self, *args, **kwargs): def append_op(self, *args, **kwargs):
op_desc = self.proto.append_op() op_desc = self.desc.append_op()
op = Operator(self, op_desc, *args, **kwargs) op = Operator(self, op_desc, *args, **kwargs)
self.ops.append(op) self.ops.append(op)
return op return op
def prepend_op(self, *args, **kwargs): def prepend_op(self, *args, **kwargs):
op_proto = self.proto.prepend_op() op_desc = self.desc.prepend_op()
op = Operator(self, op_proto, *args, **kwargs) op = Operator(self, op_desc, *args, **kwargs)
self.ops.appendleft(op) self.ops.appendleft(op)
return op return op
...@@ -162,7 +162,7 @@ class Program(object): ...@@ -162,7 +162,7 @@ class Program(object):
def __init__(self): def __init__(self):
assert not hasattr(self.__class__, assert not hasattr(self.__class__,
'_instance'), 'Do not call constructor directly!' '_instance'), 'Do not call constructor directly!'
self.proto = core.ProgramDesc.instance() self.desc = core.ProgramDesc.instance()
self.blocks = [Block(self, 0)] self.blocks = [Block(self, 0)]
self.current_block_idx = 0 self.current_block_idx = 0
...@@ -174,7 +174,7 @@ class Program(object): ...@@ -174,7 +174,7 @@ class Program(object):
def create_block(self): def create_block(self):
new_block_idx = len(self.blocks) new_block_idx = len(self.blocks)
self.proto.append_block(self.current_block().proto) self.desc.append_block(self.current_block().desc)
self.current_block_idx = new_block_idx self.current_block_idx = new_block_idx
self.blocks.append(Block(self, self.current_block_idx)) self.blocks.append(Block(self, self.current_block_idx))
return self.current_block() return self.current_block()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册