The topology on pjrt layer can be seen as: (process, chip, logical device) or (process, chip, core) For cpu, it is (1, num device, 1) For gpu, it is (num host, gpu per host, 1) PiperOrigin-RevId: 826581627
tf_vendored