Automated Code Change
Automated Code Change

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/11435 from jaro-sevcik:host-offloading-spmd-partitioning a6d01010f3b468de02ebee8a0c59f9e28ea4297d
