Allow individual layouts in the device_layouts argument of `PjRtCli…
#33304
| Job | Run time |
|---|---|
| 0s | |
| 0s |
device_layouts argument of `PjRtCli…
#33304
| Job | Run time |
|---|---|
| 0s | |
| 0s |