phivenv/Lib/site-packages/torch/distributed/tensor/device_mesh.py

from torch.distributed.device_mesh import (  # noqa: F401
    _get_device_handle,
    _mesh_resources,
    DeviceMesh,
    init_device_mesh,
)


__all__ = ["init_device_mesh", "DeviceMesh"]