| 1311 | } |
| 1312 | |
| 1313 | Result<std::shared_ptr<Buffer>> WriteRecordBatchMessage( |
| 1314 | int64_t length, int64_t body_length, |
| 1315 | const std::shared_ptr<const KeyValueMetadata>& custom_metadata, |
| 1316 | const std::vector<FieldMetadata>& nodes, const std::vector<BufferMetadata>& buffers, |
| 1317 | const std::vector<int64_t>& variadic_buffer_counts, const IpcWriteOptions& options) { |
| 1318 | FBB fbb; |
| 1319 | RecordBatchOffset record_batch; |
| 1320 | RETURN_NOT_OK(MakeRecordBatch(fbb, length, body_length, nodes, buffers, |
| 1321 | variadic_buffer_counts, options, &record_batch)); |
| 1322 | return WriteFBMessage(fbb, flatbuf::MessageHeader::MessageHeader_RecordBatch, |
| 1323 | record_batch.Union(), body_length, options.metadata_version, |
| 1324 | custom_metadata, options.memory_pool); |
| 1325 | } |
| 1326 | |
| 1327 | Result<std::shared_ptr<Buffer>> WriteTensorMessage(const Tensor& tensor, |
| 1328 | int64_t buffer_start_offset, |