| 1116 | } |
| 1117 | |
| 1118 | PyObject* GraphShape(PyObject* tensor) const { |
| 1119 | PyObject* arg_list = Py_BuildValue("(O)", tensor); |
| 1120 | PyObject* result = PyEval_CallObject(graph_shape_fn_, arg_list); |
| 1121 | Py_DECREF(arg_list); |
| 1122 | return result; |
| 1123 | } |
| 1124 | |
| 1125 | tensorflow::Status CallBackwardFunction( |
| 1126 | PyBackwardFunction* backward_function, |