Lines Matching refs:tf_record
33 from tensorflow.python.lib.io import tf_record
46 TFRecordCompressionType = tf_record.TFRecordCompressionType
595 fn = os.path.join(self.get_temp_dir(), "tf_record.%d.txt" % i)
597 writer = tf_record.TFRecordWriter(fn)
662 options = tf_record.TFRecordOptions(TFRecordCompressionType.ZLIB)
688 options = tf_record.TFRecordOptions(TFRecordCompressionType.GZIP)
715 fn = os.path.join(self.get_temp_dir(), "tf_record.%d.txt" % i)
717 options = tf_record.TFRecordOptions(
719 writer = tf_record.TFRecordWriter(fn, options=options)
727 def _WriteRecordsToFile(self, records, name="tf_record"):
729 writer = tf_record.TFRecordWriter(fn, options=None)
749 options = tf_record.TFRecordOptions(
791 options = tf_record.TFRecordOptions(
810 for r in tf_record.tf_record_iterator(
812 options=tf_record.TFRecordOptions(
813 tf_record.TFRecordCompressionType.ZLIB)):
827 for r in tf_record.tf_record_iterator(
829 options=tf_record.TFRecordOptions(
830 tf_record.TFRecordCompressionType.ZLIB)):
842 gzfn = os.path.join(self.get_temp_dir(), "tf_record.gz")
847 for r in tf_record.tf_record_iterator(
848 gzfn, options=tf_record.TFRecordOptions(TFRecordCompressionType.GZIP)):
866 compression_type=tf_record.TFRecordCompressionType.ZLIB):
868 options = tf_record.TFRecordOptions(compression_type=compression_type)
869 writer = tf_record.TFRecordWriter(fn, options=options)
888 options = tf_record.TFRecordOptions(
890 reader = tf_record.tf_record_iterator(fn, options)
904 for r in tf_record.tf_record_iterator(zfn):
914 zfn = self._ZlibDecompressFile(fn, "write_zlib_read_large.tf_record")
916 for r in tf_record.tf_record_iterator(zfn):
929 zfn = os.path.join(self.get_temp_dir(), "tf_record")
934 for r in tf_record.tf_record_iterator(zfn):
941 with tf_record.TFRecordWriter(fn) as writer:
950 for _ in tf_record.tf_record_iterator(fn_truncated):
