Add mish op to python api

openvinotoolkit · Jul 27, 2020 · 70c02bb · 70c02bb
1 parent 246451e
commit 70c02bb
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/ngraph/python/src/ngraph/__init__.py b/ngraph/python/src/ngraph/__init__.py
@@ -93,6 +93,7 @@
 from ngraph.opset4 import max_pool
 from ngraph.opset4 import maximum
 from ngraph.opset4 import minimum
+from ngraph.opset4 import mish
 from ngraph.opset4 import mod
 from ngraph.opset4 import multiply
 from ngraph.opset4 import mvn

diff --git a/ngraph/python/src/ngraph/opset4/__init__.py b/ngraph/python/src/ngraph/opset4/__init__.py
@@ -83,6 +83,7 @@
 from ngraph.opset1.ops import max_pool
 from ngraph.opset1.ops import maximum
 from ngraph.opset1.ops import minimum
+from ngraph.opset4.ops import mish
 from ngraph.opset1.ops import mod
 from ngraph.opset1.ops import multiply
 from ngraph.opset2.ops import mvn

diff --git a/ngraph/python/src/ngraph/opset4/ops.py b/ngraph/python/src/ngraph/opset4/ops.py
@@ -99,3 +99,13 @@ def non_max_suppression(
     }
 
     return _get_node_factory_opset4().create("NonMaxSuppression", inputs, attributes)
+
+
+@nameable_op
+def mish(data: NodeInput, name: Optional[str] = None,) -> Node:
+    """Return a node which performs Mish.
+
+    :param data: Tensor with input data floating point type.
+    :return: The new node which performs Mish
+    """
+    return _get_node_factory_opset4().create("Mish", as_nodes(data), {})