| 1359 | } |
| 1360 | |
| 1361 | Result<std::shared_ptr<Buffer>> WriteSparseTensorMessage( |
| 1362 | const SparseTensor& sparse_tensor, int64_t body_length, |
| 1363 | const std::vector<BufferMetadata>& buffers, const IpcWriteOptions& options) { |
| 1364 | FBB fbb; |
| 1365 | SparseTensorOffset fb_sparse_tensor; |
| 1366 | RETURN_NOT_OK( |
| 1367 | MakeSparseTensor(fbb, sparse_tensor, body_length, buffers, &fb_sparse_tensor)); |
| 1368 | return WriteFBMessage(fbb, flatbuf::MessageHeader::MessageHeader_SparseTensor, |
| 1369 | fb_sparse_tensor.Union(), body_length, options.metadata_version, |
| 1370 | /*custom_metadata=*/nullptr, options.memory_pool); |
| 1371 | } |
| 1372 | |
| 1373 | Result<std::shared_ptr<Buffer>> WriteDictionaryMessage( |
| 1374 | int64_t id, bool is_delta, int64_t length, int64_t body_length, |
no test coverage detected