Function _compute

python/tvm/topi/math.py:635–647

(*indices)

    """

    def _compute(*indices):
        value = x(*indices)
        const_min = (
            tvm.tirx.Cast(value.dtype, a_min)
            if isinstance(a_min, PrimExpr)
            else tvm.tirx.const(a_min, value.dtype)
        )
        const_max = (
            tvm.tirx.Cast(value.dtype, a_max)
            if isinstance(a_max, PrimExpr)
            else tvm.tirx.const(a_max, value.dtype)
        )
        return tvm.te.max(tvm.te.min(value, const_max), const_min)

    return te.compute(x.shape, _compute)