Add support for Sigmoid and Mul ops in PytorchParser (#510)

microsoft · Dec 6, 2018 · 72529a9 · 72529a9
1 parent eda67ef
commit 72529a9
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/mmdnn/conversion/pytorch/pytorch_parser.py b/mmdnn/conversion/pytorch/pytorch_parser.py
@@ -27,6 +27,8 @@ class PytorchParser(Parser):
     'onnx::Concat': 'Concat',
     'onnx::Relu': 'Relu',
     'onnx::Tanh': 'Tanh',
+    'onnx::Sigmoid': 'Sigmoid',
+    'onnx::Mul': 'Mul'
 
 
     # TODO
@@ -308,6 +310,12 @@ def rename_Relu(self, source_node):
     def rename_Tanh(self, source_node):
         IR_node = self._convert_identity_operation(source_node, new_op="Tanh")
 
+    def rename_Sigmoid(self, source_node):
+        IR_node = self._convert_identity_operation(source_node, new_op="Sigmoid")
+
+    def rename_Mul(self, source_node):
+        IR_node = self._convert_identity_operation(source_node, new_op="Mul")
+
     def rename_Maxpool(self, source_node):
         attr = source_node.attrs
         kwargs = dict()