xla

afd614cd - [PJRT] Count TPU chips by PCI IDs (#4845)

[PJRT] Count TPU chips by PCI IDs (#4845) * [PJRT] Count TPU chips by PCI device vendor * formatting * correct test case * Check device ID too * Update unit test