Class FloatImm

python/tvm/tirx/expr.py:581–607

Float constant. Parameters ---------- dtype : str The data type value : float The constant value. span : Optional[Span] The location of this expression in the source code.

@tvm_ffi.register_object("ir.FloatImm")
class FloatImm(ConstExpr):
    """Float constant.

    Parameters
    ----------
    dtype : str
        The data type

    value : float
        The constant value.

    span : Optional[Span]
        The location of this expression in the source code.
    """

    value: float

    def __init__(self, dtype: str, value: float, span: Span | None = None) -> None:
        self.__init_handle_by_constructor__(
            tvm.ir._ffi_api.FloatImm,
            dtype,
            value,
            span,  # type: ignore
        )

    def __float__(self) -> float:
        return self.value


@tvm_ffi.register_object("ir.IntImm")