“d006057a4c74df1aa824ff8e18a2ae24ecdefa2e”上不存在“mobile/src/operators/fusion_conv_add_relu_op.cpp”
提交 c2816a1c 编写于 作者: 刘托

Merge branch 'onnx_opt' into 'master'

opt the supported ops' info of onnx

See merge request !1166
...@@ -83,7 +83,7 @@ OnnxSupportedOps = [ ...@@ -83,7 +83,7 @@ OnnxSupportedOps = [
'Div', 'Div',
'Dropout', 'Dropout',
'DynamicLSTM', 'DynamicLSTM',
'Elu', # 'Elu',
'Equal', 'Equal',
# 'Exp', # 'Exp',
# 'Expand', # 'Expand',
...@@ -120,7 +120,7 @@ OnnxSupportedOps = [ ...@@ -120,7 +120,7 @@ OnnxSupportedOps = [
'MaxPool', 'MaxPool',
# 'MaxRoiPool', # 'MaxRoiPool',
# 'MaxUnpool', # 'MaxUnpool',
'Mean', # 'Mean',
'Min', 'Min',
'Mul', 'Mul',
# 'Multinomial', # 'Multinomial',
...@@ -371,7 +371,10 @@ class OnnxConverter(base_converter.ConverterInterface): ...@@ -371,7 +371,10 @@ class OnnxConverter(base_converter.ConverterInterface):
OnnxOpType.Relu.name: self.convert_activation, OnnxOpType.Relu.name: self.convert_activation,
OnnxOpType.Reshape.name: self.convert_reshape, OnnxOpType.Reshape.name: self.convert_reshape,
OnnxOpType.Reciprocal.name: self.convert_eltwise, OnnxOpType.Reciprocal.name: self.convert_eltwise,
OnnxOpType.ReduceMax.name: self.convert_reduce,
OnnxOpType.ReduceMean.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.Scale.name: self.convert_eltwise,
OnnxOpType.Shape.name: self.convert_shape, OnnxOpType.Shape.name: self.convert_shape,
OnnxOpType.Sigmoid.name: self.convert_activation, OnnxOpType.Sigmoid.name: self.convert_activation,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册