diff --git a/mace/python/tools/converter_tool/onnx_converter.py b/mace/python/tools/converter_tool/onnx_converter.py index 61f45e548a1c85cd26530ff41e7841978b557588..3202cb3a490a802db88587edb64af88b5ab17cb6 100644 --- a/mace/python/tools/converter_tool/onnx_converter.py +++ b/mace/python/tools/converter_tool/onnx_converter.py @@ -83,7 +83,7 @@ OnnxSupportedOps = [ 'Div', 'Dropout', 'DynamicLSTM', - 'Elu', + # 'Elu', 'Equal', # 'Exp', # 'Expand', @@ -120,7 +120,7 @@ OnnxSupportedOps = [ 'MaxPool', # 'MaxRoiPool', # 'MaxUnpool', - 'Mean', + # 'Mean', 'Min', 'Mul', # 'Multinomial', @@ -371,7 +371,10 @@ class OnnxConverter(base_converter.ConverterInterface): OnnxOpType.Relu.name: self.convert_activation, OnnxOpType.Reshape.name: self.convert_reshape, OnnxOpType.Reciprocal.name: self.convert_eltwise, + OnnxOpType.ReduceMax.name: self.convert_reduce, OnnxOpType.ReduceMean.name: self.convert_reduce, + OnnxOpType.ReduceMin.name: self.convert_reduce, + OnnxOpType.ReduceProd.name: self.convert_reduce, OnnxOpType.Scale.name: self.convert_eltwise, OnnxOpType.Shape.name: self.convert_shape, OnnxOpType.Sigmoid.name: self.convert_activation,