diff options
Diffstat (limited to 'tensorflow/core/lib/io/record_writer.h')
-rw-r--r-- | tensorflow/core/lib/io/record_writer.h | 34 |
1 files changed, 34 insertions, 0 deletions
diff --git a/tensorflow/core/lib/io/record_writer.h b/tensorflow/core/lib/io/record_writer.h new file mode 100644 index 0000000000..c7af00e5ae --- /dev/null +++ b/tensorflow/core/lib/io/record_writer.h @@ -0,0 +1,34 @@ +#ifndef TENSORFLOW_LIB_IO_RECORD_WRITER_H_ +#define TENSORFLOW_LIB_IO_RECORD_WRITER_H_ + +#include "tensorflow/core/lib/core/stringpiece.h" +#include "tensorflow/core/platform/port.h" +#include "tensorflow/core/public/status.h" + +namespace tensorflow { + +class WritableFile; + +namespace io { + +class RecordWriter { + public: + // Create a writer that will append data to "*dest". + // "*dest" must be initially empty. + // "*dest" must remain live while this Writer is in use. + explicit RecordWriter(WritableFile* dest); + + ~RecordWriter(); + + Status WriteRecord(StringPiece slice); + + private: + WritableFile* const dest_; + + TF_DISALLOW_COPY_AND_ASSIGN(RecordWriter); +}; + +} // namespace io +} // namespace tensorflow + +#endif // TENSORFLOW_LIB_IO_RECORD_WRITER_H_ |