未验证 提交 aec84219 编写于 作者: Y Yiming Wang 提交者: GitHub

add Linear and NoOp ops to support more Kaldi Components(layers) (#660)

上级 3bda6319
...@@ -108,6 +108,7 @@ OnnxSupportedOps = [ ...@@ -108,6 +108,7 @@ OnnxSupportedOps = [
'ImageScaler', 'ImageScaler',
# 'InstanceNormalization', # 'InstanceNormalization',
# 'LRN', # 'LRN',
'Linear',
'LSTM', 'LSTM',
'LstmNonlinear', 'LstmNonlinear',
'LeakyRelu', 'LeakyRelu',
...@@ -127,6 +128,7 @@ OnnxSupportedOps = [ ...@@ -127,6 +128,7 @@ OnnxSupportedOps = [
'Mul', 'Mul',
# 'Multinomial', # 'Multinomial',
'Neg', 'Neg',
'NoOp',
'Normalize', 'Normalize',
# 'Not', # 'Not',
'Offset', 'Offset',
...@@ -357,6 +359,7 @@ class OnnxConverter(base_converter.ConverterInterface): ...@@ -357,6 +359,7 @@ class OnnxConverter(base_converter.ConverterInterface):
OnnxOpType.IfDefined.name: self.convert_ifdefined, OnnxOpType.IfDefined.name: self.convert_ifdefined,
OnnxOpType.ImageScaler.name: self.convert_imagescaler, OnnxOpType.ImageScaler.name: self.convert_imagescaler,
OnnxOpType.LeakyRelu.name: self.convert_activation, OnnxOpType.LeakyRelu.name: self.convert_activation,
OnnxOpType.Linear.name: self.convert_affine,
OnnxOpType.LogSoftmax.name: self.convert_softmax, OnnxOpType.LogSoftmax.name: self.convert_softmax,
OnnxOpType.LpNormalization: self.convert_lpnormalization, OnnxOpType.LpNormalization: self.convert_lpnormalization,
OnnxOpType.LstmNonlinear.name: self.convert_lstm_nonlinear, OnnxOpType.LstmNonlinear.name: self.convert_lstm_nonlinear,
...@@ -367,6 +370,7 @@ class OnnxConverter(base_converter.ConverterInterface): ...@@ -367,6 +370,7 @@ class OnnxConverter(base_converter.ConverterInterface):
OnnxOpType.Min.name: self.convert_eltwise, OnnxOpType.Min.name: self.convert_eltwise,
OnnxOpType.Mul.name: self.convert_eltwise, OnnxOpType.Mul.name: self.convert_eltwise,
OnnxOpType.Neg.name: self.convert_eltwise, OnnxOpType.Neg.name: self.convert_eltwise,
OnnxOpType.NoOp.name: self.convert_identity,
OnnxOpType.Normalize: self.convert_normalize, OnnxOpType.Normalize: self.convert_normalize,
OnnxOpType.Offset.name: self.convert_subsample, OnnxOpType.Offset.name: self.convert_subsample,
OnnxOpType.Pad.name: self.convert_pad, OnnxOpType.Pad.name: self.convert_pad,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册