提交 5f9d410e 编写于 作者: C chengduoZH

follow comment

上级 7cd4d0ac
...@@ -418,7 +418,7 @@ class Operator(object): ...@@ -418,7 +418,7 @@ class Operator(object):
Args: Args:
block(Block): The block has the current operator. block(Block): The block has the current operator.
desc(core.OpDesc): The protobuf description of Operator. desc(core.OpDesc): The protobuf description of Operator.
type(str): The type of operator. type(str): The type of operator. Default None.
inputs(dict): The input of this Operator. it is a dictionary, for every inputs(dict): The input of this Operator. it is a dictionary, for every
element, key is the input parameter name, and value is a list of element, key is the input parameter name, and value is a list of
variables. Default None. variables. Default None.
...@@ -459,7 +459,12 @@ class Operator(object): ...@@ -459,7 +459,12 @@ class Operator(object):
'channel_recv', 'select', 'gen_nccl_id' 'channel_recv', 'select', 'gen_nccl_id'
} }
def __init__(self, block, desc, type, inputs=None, outputs=None, def __init__(self,
block,
desc,
type=None,
inputs=None,
outputs=None,
attrs=None): attrs=None):
self.block = block self.block = block
...@@ -484,7 +489,9 @@ class Operator(object): ...@@ -484,7 +489,9 @@ class Operator(object):
if len(self.desc.type()) != 0: if len(self.desc.type()) != 0:
return return
if type is None:
raise ValueError(
"`type` to initilized an Operator can not be None.")
self.desc.set_type(type) self.desc.set_type(type)
proto = OpProtoHolder.instance().get_op_proto(type) proto = OpProtoHolder.instance().get_op_proto(type)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册