| 239 | } |
| 240 | |
| 241 | Status XRTTupleAllocation::StoreToLiteral(xla::Backend* backend, |
| 242 | xla::MutableLiteralBase* literal) { |
| 243 | auto transfer_manager = backend->transfer_manager(); |
| 244 | TF_ASSIGN_OR_RETURN(auto stream, backend->BorrowStream(device_ordinal())); |
| 245 | TF_ASSIGN_OR_RETURN(xla::ShapedBuffer shaped_buffer, ToShapedBuffer()); |
| 246 | return transfer_manager->TransferLiteralFromDevice(stream.get(), |
| 247 | shaped_buffer, literal); |
| 248 | } |
| 249 | |
| 250 | Status XRTTupleAllocation::WriteLiteral(xla::Backend* backend, |
| 251 | const xla::Literal& literal) { |
nothing calls this directly
no test coverage detected