[Relax] [ONNX] Add support for Sign and Not

Author: tsu-bin

python/tvm/relax/frontend/onnx/onnx_frontend.py

@@ -1948,6 +1948,22 @@ def _impl_v14(cls, bb, inputs, attr, params):
         )
 
 
+class Sign(OnnxOpConverter):
+    """Converts an onnx Sign node into an equivalent Relax expression."""
+
+    @classmethod
+    def _impl_v9(cls, bb, inputs, attr, params):
+        return relax.op.sign(inputs[0])
+
+
+class Not(OnnxOpConverter):
+    """Converts an onnx Not node into an equivalent Relax expression."""
+
+    @classmethod
+    def _impl_v1(cls, bb, inputs, attr, params):
+        return relax.op.logical_not(inputs[0])
+
+
 def _get_convert_map():
     return {
         "MatMul": MatMul,
@@ -2030,6 +2046,8 @@ def _get_convert_map():
         "Elu": Elu,
         "HardSigmoid": HardSigmoid,
         "HardSwish": HardSwish,
+        "Sign": Sign,
+        "Not": Not,
     }
 
 

tests/python/relax/test_frontend_onnx.py

@@ -600,6 +600,14 @@ def test_hardswish():
     verify_unary("HardSwish", [32, 32])
 
 
+def test_sign():
+    verify_unary("Sign", [32, 32])
+
+
+def test_not():
+    verify_unary("Not", [32, 32], dtype=TensorProto.BOOL)
+
+
 def test_conv():
     def _verify_conv(input_shape, weight_shape, output_shape):
         bias_shape = [output_shape[1]]