diff --git a/x2paddle/op_mapper/paddle2onnx/opset9/opset.py b/x2paddle/op_mapper/paddle2onnx/opset9/opset.py index 0cec86f336db2ce8a5f0098ba6b3081ef499f5ad..83931f390f7c098bfd3f04ee2a6c6066f4b7bc4f 100644 --- a/x2paddle/op_mapper/paddle2onnx/opset9/opset.py +++ b/x2paddle/op_mapper/paddle2onnx/opset9/opset.py @@ -110,6 +110,13 @@ class OpSet9(object): 'Relu', inputs=op.input('X'), outputs=op.output('Out')) return node + def prelu(self, op, block): + node = helper.make_node( + 'PRelu', + inputs=[op.input('X')[0], op.input('Alpha')[0]], + outputs=op.output('Out')) + return node + def tanh(self, op, block): node = helper.make_node( 'Tanh', inputs=op.input('X'), outputs=op.output('Out'))