Update tfe_wrapper.cc
This commit is contained in:
parent
8cdfc53a63
commit
10ba65efab
|
@ -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,
|
||||
|
|
Loading…
Reference in New Issue