aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/lib/io/py_record_reader.i
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