Merge pull request #11287 from reyoung/feature/remove_op_role_from_doc

Do not generate doc for op_role and op_attr_name
wangkuiyi-patch-1
Yu Yang 7 years ago committed by GitHub
commit 71988fa398
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -363,6 +363,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):
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(' (')

Loading…
Cancel
Save