[JAX:TFRT:CPU] Fix TfrtCpuClient::BufferFromHostBuffer bug when shape has non-default layout (e.g., from TPU).
[JAX:TFRT:CPU] Fix TfrtCpuClient::BufferFromHostBuffer bug when shape has
non-default layout (e.g., from TPU).

PiperOrigin-RevId: 375165974

google/jax,google/jax,google/jax,google/jax