Skip to content

Commit 2dd66ef

Browse files
committed
Do not generate doc for op_role and op_attr_name
1 parent 3ff9ba0 commit 2dd66ef

File tree

2 files changed

+11
-0
lines changed

2 files changed

+11
-0
lines changed

python/paddle/fluid/framework.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -361,6 +361,13 @@ def get_op_proto(self, type):
361361
raise ValueError("Operator \"%s\" has not been registered." % type)
362362
return self.op_proto_map[type]
363363

364+
@staticmethod
365+
def generated_op_attr_names():
366+
return {
367+
core.op_proto_and_checker_maker.kOpRoleAttrName(),
368+
core.op_proto_and_checker_maker.kOpRoleVarAttrName()
369+
}
370+
364371

365372
class Operator(object):
366373
"""

python/paddle/fluid/layers/layer_function_generator.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,11 @@ def _type_to_str_(tp):
7575
buf.write(str(each_input.dispensable))
7676
buf.write('\n')
7777

78+
skip_attrs = OpProtoHolder.generated_op_attr_names()
79+
7880
for each_attr in op_proto.attrs:
81+
if each_attr.name in skip_attrs:
82+
continue
7983
buf.write(' ')
8084
buf.write(each_attr.name)
8185
buf.write(' (')

0 commit comments

Comments
 (0)