Reverts 7edfea8c24f42045c75d329ff3d498c7dd6e34b9
Reverts 7edfea8c24f42045c75d329ff3d498c7dd6e34b9

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