Method spatial

python/tvm/tirx/script/builder/ir.py:1024–1049

The spatial block axis defining function. Parameters ---------- dom : Union[Range, List[PrimExpr], Tuple[PrimExpr]] The domain of the iteration variable. binding : PrimExpr The binding value of the iteration variable. dtype : str

(
        dom: ir.Range | list[PrimExpr] | tuple[PrimExpr],
        binding: PrimExpr,
        dtype: str = "int32",
    )

    @staticmethod
    def spatial(
        dom: ir.Range | list[PrimExpr] | tuple[PrimExpr],
        binding: PrimExpr,
        dtype: str = "int32",
    ) -> Var:
        """The spatial block axis defining function.

        Parameters
        ----------
        dom : Union[Range, List[PrimExpr], Tuple[PrimExpr]]
            The domain of the iteration variable.

        binding : PrimExpr
            The binding value of the iteration variable.

        dtype : str
            The data type of the iteration variable.

        Returns
        -------
        res : Var
            The iteration variable.
        """
        return _ffi_api.AxisSpatial(  # type: ignore[attr-defined] # pylint: disable=no-member
            _as_range(dom), binding, dtype
        )

    @staticmethod
    def reduce(