Merge branch 'fix_dlpack_r2.3' of https://github.com/vovallen/tensorflow into fix_dlpack_r2.3

This commit is contained in:
VoVAllen 2020-07-02 07:37:39 +00:00
commit 4703d0ddce

View File

@ -1169,13 +1169,9 @@ PYBIND11_MODULE(_pywrap_tfe, m) {
PyCapsule_SetName(pycapsule.ptr(), "used_dltensor");
PyCapsule_SetDestructor(pycapsule.ptr(), nullptr);
<<<<<<< HEAD
return py::handle(EagerTensorFromHandle(thandle));
=======
PyObject* pyhandle = EagerTensorFromHandle(thandle);
return tensorflow::PyoOrThrow(pyhandle);
>>>>>>> ce9b1295b5... fix
});
m.def("TFE_Py_RegisterCustomDevice", [](const py::handle& context,