experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fixed a bug.

Browse Source
This commit is contained in:
Anthony Platanios 2017-08-28 16:02:47 -04:00 committed by Martin Wicke
parent 26c4a2c1ec
commit eaa69ba50b
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tensorflow/python/eager/pywrap_tfe_src.cc
View File

@ -347,8 +347,13 @@ TFE_TensorHandle* TFE_Py_NumpyToTensorHandle(PyObject* obj) {
tensorflow::Tensor t; tensorflow::Tensor t;
auto cppstatus = tensorflow::NdarrayToTensor(obj, &t); auto cppstatus = tensorflow::NdarrayToTensor(obj, &t);
if (cppstatus.ok()) { if (cppstatus.ok()) {
TFE_TensorHandle* tensor = TFE_NewTensorHandle(t, cppstatus); TF_Status* status = TF_NewStatus();
if (!cppstatus.ok()) return nullptr; TFE_TensorHandle* tensor = TFE_NewTensorHandle(t, status);
if (TF_GetCode(status) != TF_OK) {
TF_DeleteStatus(status);
return nullptr;
}
TF_DeleteStatus(status);
return tensor; return tensor;
} else { } else {
tensorflow::mutex_lock l(exception_class_mutex); tensorflow::mutex_lock l(exception_class_mutex);