Skip to content

Commit b2a78a5

Browse files
authored
add edge mode support in pading mod (#687)
1 parent d8dbf83 commit b2a78a5

File tree

1 file changed

+2
-0
lines changed
  • x2paddle/op_mapper/onnx2paddle/opset9

1 file changed

+2
-0
lines changed

x2paddle/op_mapper/onnx2paddle/opset9/opset.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -518,6 +518,8 @@ def Pad(self, node, op_independent=True):
518518
if pads is not None:
519519
is_pads_attr = True
520520
mode = node.get_attr('mode', 'constant')
521+
if mode in ["edge"]:
522+
mode = "replicate"
521523
value = node.get_attr('value', 0.)
522524
data_shape = val_x.out_shapes[0]
523525
output_shape = node.out_shapes[0]

0 commit comments

Comments
 (0)