| 1460 | } |
| 1461 | |
| 1462 | Status SegmentBundleWriter::End() { |
| 1463 | if (write_counter_ * buffer_size_ + buffer_ptr_ != |
| 1464 | shape_.num_elements() * DataTypeSize(type_)) { |
| 1465 | return errors::Internal("SegmentBundleWriter write size error"); |
| 1466 | } |
| 1467 | if (write_counter_ == 0) { |
| 1468 | return writer_->AddCompeleteData(buffer_.get(), buffer_ptr_); |
| 1469 | } else if (buffer_ptr_ > 0) { |
| 1470 | TF_RETURN_IF_ERROR(writer_->AppendSegmentData(buffer_.get(), buffer_ptr_)); |
| 1471 | writer_->EndSegmentData( |
| 1472 | write_counter_ * buffer_size_ + buffer_ptr_, buffer_ptr_); |
| 1473 | return Status::OK(); |
| 1474 | } else { |
| 1475 | writer_->EndSegmentData( |
| 1476 | write_counter_ * buffer_size_ + buffer_ptr_, buffer_size_); |
| 1477 | return Status::OK(); |
| 1478 | } |
| 1479 | } |
| 1480 | |
| 1481 | SegmentBundleReader::SegmentBundleReader( |
| 1482 | BundleReader* reader, const string& name, |
nothing calls this directly
no test coverage detected