Function fast_exp

python/tvm/topi/math.py:788–803

Take exponential of input x using fast_exp implementation Parameters ---------- x : tvm.te.Tensor Input argument. Returns ------- y : tvm.te.Tensor The result.

(x)

def fast_exp(x):
    """Take exponential of input x using fast_exp implementation

    Parameters
    ----------
    x : tvm.te.Tensor
        Input argument.

    Returns
    -------
    y : tvm.te.Tensor
        The result.
    """
    if x.dtype.startswith("int") or x.dtype.startswith("uint"):
        x = cast(x, "float32")
    return cpp.fast_exp(x, x.dtype, tag.ELEMWISE)


def fast_tanh(x):