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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
// SWIG wrapper for lib::tensorflow::Status
%include "tensorflow/python/platform/base.i"
%include "tensorflow/python/lib/core/strings.i"
%apply int { tensorflow::error::Code }; // Treat the enum as an integer.
%{
#include "tensorflow/core/public/status.h"
%}
%typemap(out, fragment="StatusNotOK") tensorflow::Status {
if ($1.ok()) {
$result = SWIG_Py_Void();
} else {
RaiseStatusNotOK($1, $descriptor(tensorflow::Status*));
SWIG_fail;
}
}
%init %{
// Setup the StatusNotOK exception class.
PyObject *pywrap_status = PyImport_ImportModuleNoBlock(
"tensorflow.python.pywrap_tensorflow");
if (pywrap_status) {
PyObject *exception = PyErr_NewException(
"tensorflow.python.pywrap_tensorflow.StatusNotOK",
NULL, NULL);
if (exception) {
PyModule_AddObject(pywrap_status, "StatusNotOK", exception); // Steals ref.
}
Py_DECREF(pywrap_status);
}
%}
%fragment("StatusNotOK", "header") %{
#include "tensorflow/core/public/status.h"
namespace {
// Initialized on the first call to RaiseStatusNotOK().
static PyObject *StatusNotOKError = nullptr;
inline void Py_DECREF_wrapper(PyObject *o) { Py_DECREF(o); }
typedef std::unique_ptr<PyObject, decltype(&Py_DECREF_wrapper)> SafePyObjectPtr;
SafePyObjectPtr make_safe(PyObject* o) {
return SafePyObjectPtr(o, Py_DECREF_wrapper);
}
void RaiseStatusNotOK(const tensorflow::Status& status, swig_type_info *type) {
const int code = status.code();
string fullmsg = status.ToString();
PyObject *exception = nullptr;
// We're holding the Python GIL, so we don't need to synchronize
// access to StatusNotOKError with a Mutex of our own.
if (!StatusNotOKError) {
PyObject *cls = nullptr;
auto pywrap = make_safe(PyImport_ImportModule(
"tensorflow.python.pywrap_tensorflow"));
if (pywrap) {
cls = PyObject_GetAttrString(pywrap.get(), "StatusNotOK");
}
if (!cls) {
cls = Py_None;
Py_INCREF(cls);
}
StatusNotOKError = cls;
}
if (StatusNotOKError != Py_None) {
auto fullmsg_ptr = make_safe(_SwigString_FromString(fullmsg));
auto exception_ptr = make_safe(PyObject_CallFunctionObjArgs(
StatusNotOKError, fullmsg_ptr.get(), NULL));
exception = exception_ptr.get();
if (exception) {
auto pycode = make_safe(PyInt_FromLong(static_cast<long>(code)));
auto pymsg = make_safe(_SwigString_FromString(status.error_message()));
auto pystatus = make_safe(SWIG_NewPointerObj(
SWIG_as_voidptr(new tensorflow::Status(status)), type, SWIG_POINTER_OWN));
PyObject_SetAttrString(exception, "code", pycode.get());
PyObject_SetAttrString(exception, "error_message", pymsg.get());
PyErr_SetObject(StatusNotOKError, exception);
}
}
if (!exception) {
fullmsg =
("could not construct StatusNotOK (original error "
" was: " +
fullmsg + ")");
PyErr_SetString(PyExc_SystemError, fullmsg.c_str());
}
}
} // namespace
%}
%ignoreall
%unignore tensorflow;
%unignore tensorflow::lib;
%unignore tensorflow::Status;
%unignore tensorflow::Status::Status;
%unignore tensorflow::Status::Status(tensorflow::error::Code, StringPiece);
%unignore tensorflow::Status::~Status;
%unignore tensorflow::Status::code;
%unignore tensorflow::Status::ok;
%unignore tensorflow::Status::error_message;
%unignore tensorflow::Status::ToString;
%ignore tensorflow::Status::operator=;
%rename(__str__) tensorflow::Status::ToString;
%include "tensorflow/core/public/status.h"
%unignoreall
|
