From a8bc1b610ee458263d674e0b533e64d21f96e618 Mon Sep 17 00:00:00 2001
From: Channingss <chen_lingchi@163.com>
Date: Tue, 7 Jul 2020 09:44:03 +0000
Subject: [PATCH] [ONNX] add Greater, ReduceMax

---
 x2paddle/op_mapper/onnx_opsets/opset9.py | 20 ++++++++++++++++----
 1 file changed, 16 insertions(+), 4 deletions(-)

diff --git a/x2paddle/op_mapper/onnx_opsets/opset9.py b/x2paddle/op_mapper/onnx_opsets/opset9.py
index 5cf058f..931900b 100644
--- a/x2paddle/op_mapper/onnx_opsets/opset9.py
+++ b/x2paddle/op_mapper/onnx_opsets/opset9.py
@@ -107,6 +107,10 @@ class OpSet9():
             'reduce_min', ['X'], ['Out'], dict(
                 axes='dim', keepdims='keep_dim'), dict(keep_dim=1)
         ],
+        'ReduceMax': [
+            'reduce_max', ['X'], ['Out'], dict(
+                axes='dim', keepdims='keep_dim'), dict(keep_dim=1)
+        ],
         #active function
         'Relu': ['relu', ['X'], ['Out']],
         'LeakyRelu': ['leaky_relu', ['X'], ['Out'], dict(), dict(alpha=.01)],
@@ -131,10 +135,7 @@ class OpSet9():
         'Abs': ['abs', ['X'], ['Out']],
     }
 
-    default_ioa_constraint = {
-        'Gather':
-        [(lambda i, o, a: a.get('axis', 0) == 0, 'only axis = 0 is supported')],
-    }
+    default_ioa_constraint = {}
 
     def __init__(self, decoder):
         super(OpSet9, self).__init__()
@@ -1082,6 +1083,17 @@ class OpSet9():
             output=node,
             param_attr=None)
 
+    @print_mapping_info
+    def Greater(self, node):
+        val_x = self.graph.get_input_node(node, idx=0, copy=True)
+        val_y = self.graph.get_input_node(node, idx=1, copy=True)
+        node.fluid_code.add_layer(
+            "greater_than",
+            inputs={'x': val_x,
+                    'y': val_y},
+            output=node,
+            param_attr=None)
+
     @print_mapping_info
     def Where(self, node):
         condition = self.graph.get_input_node(node, idx=0, copy=True)
-- 
GitLab