diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-06-27 16:33:00 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-06-27 16:37:09 -0700 |
| commit | 50b999a8336d19400ab75aea66fe46eca2f5fe0b (patch) | |
| tree | 7cba4f4af6b131c253b65ff9f2923e851184668c /tensorflow/python/lib/io/tf_record.py | |
| parent | d6d58a3a1785785679af56c0f8f131e7312b8226 (diff) | |
Merge changes from github.
PiperOrigin-RevId: 160344052
Diffstat (limited to 'tensorflow/python/lib/io/tf_record.py')
| -rw-r--r-- | tensorflow/python/lib/io/tf_record.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/python/lib/io/tf_record.py b/tensorflow/python/lib/io/tf_record.py index 3d0cdc2153..df19010068 100644 --- a/tensorflow/python/lib/io/tf_record.py +++ b/tensorflow/python/lib/io/tf_record.py @@ -121,6 +121,11 @@ class TFRecordWriter(object): """ self._writer.WriteRecord(record) + def flush(self): + """Flush the file.""" + with errors.raise_exception_on_not_ok_status() as status: + self._writer.Flush(status) + def close(self): """Close the file.""" with errors.raise_exception_on_not_ok_status() as status: |