diff options
author | Alexandre Passos <apassos@google.com> | 2017-03-27 10:49:46 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-03-27 12:09:43 -0700 |
commit | 11bb09b5eb35414e1fca8899775bbda48a658040 (patch) | |
tree | df0e9c8482613f6304ebfff51528195ef043dfa1 | |
parent | 00e7f04c2d5ea9edc6a28ddeb80b4cd243bf4c40 (diff) |
Forgotten unref in py_func.
Change: 151349771
-rw-r--r-- | tensorflow/python/lib/core/py_func.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/lib/core/py_func.cc b/tensorflow/python/lib/core/py_func.cc index 217db291e7..fd9c64939d 100644 --- a/tensorflow/python/lib/core/py_func.cc +++ b/tensorflow/python/lib/core/py_func.cc @@ -316,6 +316,7 @@ Status ConvertNdarrayToTensor(PyObject* obj, Tensor* ret) { input, shape.num_elements() * DataTypeSize(dtype), PyArray_DATA(input)); *ret = buf->MakeTensor(dtype, shape); + buf->Unref(); } } } |