| 49 | } |
| 50 | |
| 51 | void PyRecordWriter::WriteRecord(tensorflow::StringPiece record, |
| 52 | TF_Status* out_status) { |
| 53 | if (writer_ == nullptr) { |
| 54 | TF_SetStatus(out_status, TF_FAILED_PRECONDITION, |
| 55 | "Writer not initialized or previously closed"); |
| 56 | return; |
| 57 | } |
| 58 | Status s = writer_->WriteRecord(record); |
| 59 | if (!s.ok()) { |
| 60 | Set_TF_Status_from_Status(out_status, s); |
| 61 | } |
| 62 | } |
| 63 | |
| 64 | void PyRecordWriter::Flush(TF_Status* out_status) { |
| 65 | if (writer_ == nullptr) { |
no test coverage detected