blob: a8c1b0dd5dd4a695eecc868bd307938cc645319a (
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
|
#ifndef TENSORFLOW_LIB_IO_RECORD_READER_H_
#define TENSORFLOW_LIB_IO_RECORD_READER_H_
#include "tensorflow/core/lib/core/stringpiece.h"
#include "tensorflow/core/platform/port.h"
#include "tensorflow/core/public/status.h"
namespace tensorflow {
class RandomAccessFile;
namespace io {
class RecordReader {
public:
// Create a reader that will return log records from "*file".
// "*file" must remain live while this Reader is in use.
explicit RecordReader(RandomAccessFile* file);
~RecordReader();
// Read the record at "*offset" into *record and update *offset to
// point to the offset of the next record. Returns OK on success,
// OUT_OF_RANGE for end of file, or something else for an error.
Status ReadRecord(uint64* offset, string* record);
private:
RandomAccessFile* src_;
TF_DISALLOW_COPY_AND_ASSIGN(RecordReader);
};
} // namespace io
} // namespace tensorflow
#endif // TENSORFLOW_LIB_IO_RECORD_READER_H_
|
