Move host-compute-offloading wrapping after SPMD sharding to shard calls executed on host
Move host-compute-offloading wrapping after SPMD sharding to shard calls executed on host

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/12915 from openxla:aportnoy/ffi-attribute-map-handle-unsigned-integers 7a26d6fa5ca122ad3ad56ba9e3668eab8c9b0235
