(x, device)
| 57 | |
| 58 | |
| 59 | def _device_put_impl(x, device): |
| 60 | x = canonicalize_arg(x) |
| 61 | return device_put_handlers[type(x)](x, device) |
| 62 | |
| 63 | |
| 64 | def device_put(x, devices: Sequence[xb.xla_client.Device], replicate: bool = False): |
no test coverage detected