unary

Author: xadupre

_unittests/ut_reference/test_torch_evaluator.py

@@ -88,13 +88,16 @@ def test_op_binary_cmp(self):
         model = oh.make_model(
             oh.make_graph(
                 [
-                    oh.make_node("Greater", ["X", "Y"], ["a"]),
+                    oh.make_node("Neg", ["X"], ["nx"]),
+                    oh.make_node("Reciprocal", ["nx"], ["rnx"]),
+                    oh.make_node("Greater", ["X", "rnx"], ["a"]),
                     oh.make_node("GreaterOrEqual", ["X", "Y"], ["b"]),
                     oh.make_node("Less", ["X", "Y"], ["c"]),
                     oh.make_node("LessOrEqual", ["X", "Y"], ["d"]),
                     oh.make_node("And", ["a", "b"], ["ab"]),
                     oh.make_node("Or", ["c", "d"], ["cd"]),
-                    oh.make_node("And", ["ab", "cd"], ["Z"]),
+                    oh.make_node("Not", ["cd"], ["ncd"]),
+                    oh.make_node("And", ["ab", "ncd"], ["Z"]),
                 ],
                 "dummy",
                 [

onnx_diagnostic/reference/torch_ops/__init__.py

@@ -16,3 +16,4 @@
 from .other_ops import Cast_6, Concat_1, Transpose_1
 from .nn_ops import Softmax_13, Tanh_6
 from .shape_ops import Reshape_14, Shape_15, Squeeze_13, Unsqueeze_13
+from .unary_ops import Neg_1, Not_1, Reciprocal_1

onnx_diagnostic/reference/torch_ops/unary_ops.py

@@ -0,0 +1,22 @@
+from . import OpRun, OpRunValue
+
+
+class Neg_1(OpRun):
+    """Neg"""
+
+    def run(self, x: OpRunValue) -> OpRunValue:
+        return OpRunValue(-x.tensor)
+
+
+class Not_1(OpRun):
+    """Not"""
+
+    def run(self, x: OpRunValue) -> OpRunValue:
+        return OpRunValue(~x.tensor)
+
+
+class Reciprocal_1(OpRun):
+    """REciprocal"""
+
+    def run(self, x: OpRunValue) -> OpRunValue:
+        return OpRunValue(1 / x.tensor)