提交 367a2b07 编写于 作者: W wjj19950828

fixed pad op

上级 9a6c85a5
...@@ -630,7 +630,7 @@ class OpSet9(): ...@@ -630,7 +630,7 @@ class OpSet9():
layer_attrs["pad"] = paddings layer_attrs["pad"] = paddings
paddle_op = "custom_layer:PadAllDim4WithOneInput" paddle_op = "custom_layer:PadAllDim4WithOneInput"
else: else:
pad_data = node.get_attr('pads') pad_data = pads
pad_data1 = pad_data[0::2] pad_data1 = pad_data[0::2]
pad_data_all = [] pad_data_all = []
for i in range(len(pad_data1)): for i in range(len(pad_data1)):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册