Function redistribute

python/tvm/relax/script/builder/distributed/ir.py:163–170

(
    value: Expr, device_mesh: py_str | DeviceMesh, placement: py_str | Placement
)

def redistribute(
    value: Expr, device_mesh: py_str | DeviceMesh, placement: py_str | Placement
) -> Expr:
    if isinstance(device_mesh, py_str):
        device_mesh = _lookup_device_mesh(device_mesh)
    if isinstance(placement, py_str):
        placement = Placement.from_text(placement)
    return _redistribute(value, device_mesh, placement)