| 33 | namespace io { |
| 34 | |
| 35 | class RecordReaderOptions { |
| 36 | public: |
| 37 | enum CompressionType { NONE = 0, ZLIB_COMPRESSION = 1 }; |
| 38 | CompressionType compression_type = NONE; |
| 39 | |
| 40 | // If buffer_size is non-zero, then all reads must be sequential, and no |
| 41 | // skipping around is permitted. (Note: this is the same behavior as reading |
| 42 | // compressed files.) Consider using SequentialRecordReader. |
| 43 | int64 buffer_size = 0; |
| 44 | |
| 45 | static RecordReaderOptions CreateRecordReaderOptions( |
| 46 | const string& compression_type); |
| 47 | |
| 48 | #if !defined(IS_SLIM_BUILD) |
| 49 | // Options specific to zlib compression. |
| 50 | ZlibCompressionOptions zlib_options; |
| 51 | #endif // IS_SLIM_BUILD |
| 52 | }; |
| 53 | |
| 54 | // Low-level interface to read TFRecord files. |
| 55 | // |
no outgoing calls