blob: e3fd05bd9a2f319c0d31063cba583d6c38a66e1e (
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
35
36
37
38
|
#ifndef THIRD_PARTY_TENSORFLOW_PYTHON_LIB_IO_PY_RECORD_WRITER_H_
#define THIRD_PARTY_TENSORFLOW_PYTHON_LIB_IO_PY_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;
// A wrapper around io::RecordWriter that is more easily SWIG wrapped for
// Python. An instance of this class is not safe for concurrent access
// by multiple threads.
class PyRecordWriter {
public:
static PyRecordWriter* New(const string& filename);
~PyRecordWriter();
bool WriteRecord(::tensorflow::StringPiece record);
void Close();
private:
PyRecordWriter();
WritableFile* file_; // Owned
io::RecordWriter* writer_; // Owned
TF_DISALLOW_COPY_AND_ASSIGN(PyRecordWriter);
};
} // namespace io
} // namespace tensorflow
#endif // THIRD_PARTY_TENSORFLOW_PYTHON_LIB_IO_PY_RECORD_WRITER_H_
|