提交 adb6b1b9 编写于 作者: L luxuhui

opt the supported ops' info of onnx

N/A
Signed-off-by: NLuxuhui <luxuhui@xiaomi.com>
上级 18088a9f
......@@ -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,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册