未验证 提交 b2a78a5f 编写于 作者: Y yeliang2258 提交者: GitHub

add edge mode support in pading mod (#687)

上级 d8dbf834
...@@ -518,6 +518,8 @@ class OpSet9(): ...@@ -518,6 +518,8 @@ class OpSet9():
if pads is not None: if pads is not None:
is_pads_attr = True is_pads_attr = True
mode = node.get_attr('mode', 'constant') mode = node.get_attr('mode', 'constant')
if mode in ["edge"]:
mode = "replicate"
value = node.get_attr('value', 0.) value = node.get_attr('value', 0.)
data_shape = val_x.out_shapes[0] data_shape = val_x.out_shapes[0]
output_shape = node.out_shapes[0] output_shape = node.out_shapes[0]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册