diff options
Diffstat (limited to 'tensorflow/python/lib/io/py_record_writer.h')
| -rw-r--r-- | tensorflow/python/lib/io/py_record_writer.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/lib/io/py_record_writer.h b/tensorflow/python/lib/io/py_record_writer.h index 8c53420ce6..9d66c031d4 100644 --- a/tensorflow/python/lib/io/py_record_writer.h +++ b/tensorflow/python/lib/io/py_record_writer.h @@ -44,6 +44,7 @@ class PyRecordWriter { ~PyRecordWriter(); bool WriteRecord(tensorflow::StringPiece record); + void Flush(TF_Status* out_status); void Close(TF_Status* out_status); private: |