From aec8421951390ee1dfec4db5c118805908776330 Mon Sep 17 00:00:00 2001
From: Yiming Wang <freewym@gmail.com>
Date: Fri, 3 Jul 2020 04:42:01 -0400
Subject: [PATCH] add Linear and NoOp ops to support more Kaldi
 Components(layers) (#660)

---
 tools/python/transform/onnx_converter.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tools/python/transform/onnx_converter.py b/tools/python/transform/onnx_converter.py
index 8a85c105..8f0b485b 100644
--- a/tools/python/transform/onnx_converter.py
+++ b/tools/python/transform/onnx_converter.py
@@ -108,6 +108,7 @@ OnnxSupportedOps = [
     'ImageScaler',
     # 'InstanceNormalization',
     # 'LRN',
+    'Linear',
     'LSTM',
     'LstmNonlinear',
     'LeakyRelu',
@@ -127,6 +128,7 @@ OnnxSupportedOps = [
     'Mul',
     # 'Multinomial',
     'Neg',
+    'NoOp',
     'Normalize',
     # 'Not',
     'Offset',
@@ -357,6 +359,7 @@ class OnnxConverter(base_converter.ConverterInterface):
             OnnxOpType.IfDefined.name: self.convert_ifdefined,
             OnnxOpType.ImageScaler.name: self.convert_imagescaler,
             OnnxOpType.LeakyRelu.name: self.convert_activation,
+            OnnxOpType.Linear.name: self.convert_affine,
             OnnxOpType.LogSoftmax.name: self.convert_softmax,
             OnnxOpType.LpNormalization: self.convert_lpnormalization,
             OnnxOpType.LstmNonlinear.name: self.convert_lstm_nonlinear,
@@ -367,6 +370,7 @@ class OnnxConverter(base_converter.ConverterInterface):
             OnnxOpType.Min.name: self.convert_eltwise,
             OnnxOpType.Mul.name: self.convert_eltwise,
             OnnxOpType.Neg.name: self.convert_eltwise,
+            OnnxOpType.NoOp.name: self.convert_identity,
             OnnxOpType.Normalize: self.convert_normalize,
             OnnxOpType.Offset.name: self.convert_subsample,
             OnnxOpType.Pad.name: self.convert_pad,
-- 
GitLab