提交 2dd66ef6 编写于 作者: Y yuyang18

Do not generate doc for op_role and op_attr_name

上级 3ff9ba0e
...@@ -361,6 +361,13 @@ class OpProtoHolder(object): ...@@ -361,6 +361,13 @@ class OpProtoHolder(object):
raise ValueError("Operator \"%s\" has not been registered." % type) raise ValueError("Operator \"%s\" has not been registered." % type)
return self.op_proto_map[type] return self.op_proto_map[type]
@staticmethod
def generated_op_attr_names():
return {
core.op_proto_and_checker_maker.kOpRoleAttrName(),
core.op_proto_and_checker_maker.kOpRoleVarAttrName()
}
class Operator(object): class Operator(object):
""" """
......
...@@ -75,7 +75,11 @@ def _generate_doc_string_(op_proto): ...@@ -75,7 +75,11 @@ def _generate_doc_string_(op_proto):
buf.write(str(each_input.dispensable)) buf.write(str(each_input.dispensable))
buf.write('\n') buf.write('\n')
skip_attrs = OpProtoHolder.generated_op_attr_names()
for each_attr in op_proto.attrs: for each_attr in op_proto.attrs:
if each_attr.name in skip_attrs:
continue
buf.write(' ') buf.write(' ')
buf.write(each_attr.name) buf.write(each_attr.name)
buf.write(' (') buf.write(' (')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册