| 1089 | } // namespace internal |
| 1090 | |
| 1091 | Status WriteSparseTensor(const SparseTensor& sparse_tensor, io::OutputStream* dst, |
| 1092 | int32_t* metadata_length, int64_t* body_length) { |
| 1093 | IpcPayload payload; |
| 1094 | internal::SparseTensorSerializer writer(0, &payload); |
| 1095 | RETURN_NOT_OK(writer.Assemble(sparse_tensor)); |
| 1096 | |
| 1097 | *body_length = payload.body_length; |
| 1098 | return WriteIpcPayload(payload, IpcWriteOptions::Defaults(), dst, metadata_length); |
| 1099 | } |
| 1100 | |
| 1101 | Status GetSparseTensorPayload(const SparseTensor& sparse_tensor, MemoryPool* pool, |
| 1102 | IpcPayload* out) { |