Throw an exception on failure to read records in tf_record_iterator

Change: 140506546

1 files changed, 9 insertions, 1 deletions

diff --git a/tensorflow/python/summary/impl/event_file_loader.py b/tensorflow/python/summary/impl/event_file_loader.py
index dedebe5484..ccc61d4564 100644
--- a/tensorflow/python/summary/impl/event_file_loader.py
+++ b/tensorflow/python/summary/impl/event_file_loader.py
@@ -52,7 +52,15 @@ class EventFileLoader(object):
     Yields:
       All values that were written to disk that have not been yielded yet.
     """
-    while self._reader.GetNext():
+    while True:
+      try:
+        with errors.raise_exception_on_not_ok_status() as status:
+          self._reader.GetNext(status)
+      except (errors.DataLossError, errors.OutOfRangeError):
+        # We ignore partial read exceptions, because a record may be truncated.
+        # PyRecordReader holds the offset prior to the failed read, so retrying
+        # will succeed.
+        break
       event = event_pb2.Event()
       event.ParseFromString(self._reader.record())
       yield event