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 |