Allow individual layouts in the device_layouts
argument of `PjRtCli…
#14710
Job | Run time |
---|---|
1m 12s | |
1m 12s |
device_layouts
argument of `PjRtCli…
#14710
Job | Run time |
---|---|
1m 12s | |
1m 12s |