diff options
Diffstat (limited to 'tensorflow/python/lib/io/py_record_reader.i')
-rw-r--r-- | tensorflow/python/lib/io/py_record_reader.i | 39 |
1 files changed, 39 insertions, 0 deletions
diff --git a/tensorflow/python/lib/io/py_record_reader.i b/tensorflow/python/lib/io/py_record_reader.i new file mode 100644 index 0000000000..19f911bd52 --- /dev/null +++ b/tensorflow/python/lib/io/py_record_reader.i @@ -0,0 +1,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 |