[paddle2onnx] add prelu

422412a0 · channingss · 2c12e94b · 422412a0
隐藏空白更改
内联并排

Showing with 7 addition and 0 deletion

x2paddle/op_mapper/paddle2onnx/opset9/opset.py x2paddle/op_mapper/paddle2onnx/opset9/opset.py +7 -0

未找到文件。
--- 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'))