blob: 19f911bd52d46c533745b35e2b60573a8ba5f6c2 (
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
39
|
%nothread tensorflow::io::PyRecordReader::GetNext;
%include "tensorflow/python/platform/base.i"
%feature("except") tensorflow::io::PyRecordReader::New {
// Let other threads run while we read
Py_BEGIN_ALLOW_THREADS
$action
Py_END_ALLOW_THREADS
}
%newobject tensorflow::io::PyRecordReader::New;
%feature("except") tensorflow::io::PyRecordReader::GetNext {
// Let other threads run while we read
Py_BEGIN_ALLOW_THREADS
$action
Py_END_ALLOW_THREADS
}
%{
#include "tensorflow/python/lib/io/py_record_reader.h"
%}
%ignoreall
%unignore tensorflow;
%unignore tensorflow::io;
%unignore tensorflow::io::PyRecordReader;
%unignore tensorflow::io::PyRecordReader::~PyRecordReader;
%unignore tensorflow::io::PyRecordReader::GetNext;
%unignore tensorflow::io::PyRecordReader::offset;
%unignore tensorflow::io::PyRecordReader::record;
%unignore tensorflow::io::PyRecordReader::Close;
%unignore tensorflow::io::PyRecordReader::New;
%include "tensorflow/python/lib/io/py_record_reader.h"
%unignoreall
|