xla

[PJRT|TPU] Update `test_xla_devices_single_process_all_chips` for expected device number

#5421

Merged