提交 439a2657 编写于 作者: Y yuyang18

Better trim dot

上级 b000e0de
...@@ -736,7 +736,7 @@ def max_sequence_len(rank_table): ...@@ -736,7 +736,7 @@ def max_sequence_len(rank_table):
rank_table(${rank_table_type}): ${rank_table_comment}. rank_table(${rank_table_type}): ${rank_table_comment}.
Returns: Returns:
(${out_type}): ${out_comment} (${out_type}): ${out_comment}.
""" """
helper = LayerHelper("max_seqence_len", **locals()) helper = LayerHelper("max_seqence_len", **locals())
res = helper.create_tmp_variable(dtype="int64") res = helper.create_tmp_variable(dtype="int64")
......
...@@ -238,6 +238,9 @@ def templatedoc(): ...@@ -238,6 +238,9 @@ def templatedoc():
Decorated function. Decorated function.
""" """
def trim_ending_dot(msg):
return msg.rstrip('.')
def __impl__(func): def __impl__(func):
op_proto = OpProtoHolder.instance().get_op_proto(func.__name__) op_proto = OpProtoHolder.instance().get_op_proto(func.__name__)
tmpl = string.Template(func.__doc__) tmpl = string.Template(func.__doc__)
...@@ -249,19 +252,22 @@ def templatedoc(): ...@@ -249,19 +252,22 @@ def templatedoc():
comment += line comment += line
comment += "\n" comment += "\n"
args = {"comment": comment} args = {"comment": trim_ending_dot(comment)}
for each_input in op_proto.inputs: for each_input in op_proto.inputs:
input_name = _convert_(each_input.name) input_name = _convert_(each_input.name)
args["{0}_comment".format(input_name)] = each_input.comment args["{0}_comment".format(input_name)] = trim_ending_dot(
each_input.comment)
args["{0}_type".format(input_name)] = "Variable" args["{0}_type".format(input_name)] = "Variable"
for each_attr in op_proto.attrs: for each_attr in op_proto.attrs:
input_name = _convert_(each_attr.name) input_name = _convert_(each_attr.name)
args["{0}_comment".format(input_name)] = each_attr.comment args["{0}_comment".format(input_name)] = trim_ending_dot(
each_attr.comment)
args["{0}_type".format(input_name)] = _type_to_str_(each_attr.type) args["{0}_type".format(input_name)] = _type_to_str_(each_attr.type)
for each_opt in op_proto.outputs: for each_opt in op_proto.outputs:
output_name = _convert_(each_opt.name) output_name = _convert_(each_opt.name)
args["{0}_comment".format(output_name)] = each_opt.comment args["{0}_comment".format(output_name)] = trim_ending_dot(
each_opt.comment)
args["{0}_type".format(output_name)] = "Variable" args["{0}_type".format(output_name)] = "Variable"
func.__doc__ = tmpl.substitute(args) func.__doc__ = tmpl.substitute(args)
return func return func
......
...@@ -296,7 +296,7 @@ def fill_constant_batch_size_like(input, ...@@ -296,7 +296,7 @@ def fill_constant_batch_size_like(input,
output_dim_idx(${output_dim_idx_type}): ${output_dim_idx_comment}. output_dim_idx(${output_dim_idx_type}): ${output_dim_idx_comment}.
Returns: Returns:
${out_comment} ${out_comment}.
""" """
helper = LayerHelper("fill_constant_batch_size_like", **locals()) helper = LayerHelper("fill_constant_batch_size_like", **locals())
out = helper.create_tmp_variable(dtype=dtype) out = helper.create_tmp_variable(dtype=dtype)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册