blob: c7af00e5ae486cf01b408c5d8696a22181e03dd4 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
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_
|