[Onnx Operators] Celu (apache#8741)

* complete celu op

* forgot to add test

* change order in convert_map, remove comment, delete import hiccup

Co-authored-by: CircleSpin <jocelyn@pop-os.localdomain>

python/tvm/relay/frontend/onnx.py

@@ -3480,6 +3480,7 @@ def _get_convert_map(opset):
         "IsNaN": Renamer("isnan"),
         "Sqrt": Renamer("sqrt"),
         "Relu": Renamer("relu"),
+        "Celu": Celu.get_converter(opset),
         "LeakyRelu": Renamer("leaky_relu"),
         "Selu": Selu.get_converter(opset),
         "Elu": Elu.get_converter(opset),
@@ -3925,6 +3926,19 @@ def _fix_outputs(self, op_name, outputs):
         return outputs
 
 
+class Celu(OnnxOpConverter):
+    """Operator convereter for celu"""
+
+    @classmethod
+    def _impl_v12(cls, inputs, attr, params):
+        x = inputs[0]
+        dtype = infer_type(x).checked_type.dtype
+        alpha = _op.const(attr.get("alpha", 1.0), dtype)
+        zero = _op.const(0, dtype)
+        one = _op.const(1, dtype)
+        return _op.maximum(zero, x) + _op.minimum(zero, alpha * (_op.exp(x / alpha) - one))
+
+
 def from_onnx(
     model, shape=None, dtype="float32", opset=None, freeze_params=False, convert_config=None
 ):

tests/python/frontend/onnx/test_forward.py

@@ -4668,7 +4668,6 @@ def verify_eyelike(indata):
     "test_cast_FLOAT_to_BFLOAT16",
     "test_cast_FLOAT_to_STRING",
     "test_cast_STRING_to_FLOAT",
-    "test_celu",
     "test_compress_0",
     "test_compress_1",
     "test_compress_default_axis",