| 35 | namespace io { |
| 36 | |
| 37 | class RecordWriterOptions { |
| 38 | public: |
| 39 | enum CompressionType { NONE = 0, ZLIB_COMPRESSION = 1 }; |
| 40 | CompressionType compression_type = NONE; |
| 41 | |
| 42 | static RecordWriterOptions CreateRecordWriterOptions( |
| 43 | const string& compression_type); |
| 44 | |
| 45 | // Options specific to zlib compression. |
| 46 | #if !defined(IS_SLIM_BUILD) |
| 47 | tensorflow::io::ZlibCompressionOptions zlib_options; |
| 48 | #endif // IS_SLIM_BUILD |
| 49 | }; |
| 50 | |
| 51 | class RecordWriter { |
| 52 | public: |
no outgoing calls