| 393 | } |
| 394 | |
| 395 | Status LocalClient::TransferToInfeedLocal(const LiteralSlice& literal, |
| 396 | int device_ordinal) { |
| 397 | TF_ASSIGN_OR_RETURN(se::StreamExecutor * executor, |
| 398 | backend().stream_executor(device_ordinal)); |
| 399 | return backend().transfer_manager()->TransferLiteralToInfeed(executor, |
| 400 | literal); |
| 401 | } |
| 402 | |
| 403 | StatusOr<Literal> LocalClient::TransferFromOutfeedLocal(const Shape& shape, |
| 404 | int device_ordinal) { |