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

[Lite] data: Fix memory leak

Browse Source
This commit is contained in:
Gaurav Singh 2020-05-02 14:50:25 -04:00
parent 25b0e7d995
commit 87fc837920
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/lite/python/interpreter_wrapper/interpreter_wrapper.cc
View File

@ -592,6 +592,7 @@ PyObject* InterpreterWrapper::GetTensor(int i) const {
size_t size_of_type; size_t size_of_type;
if (GetSizeOfType(nullptr, tensor->type, &size_of_type) != kTfLiteOk) { if (GetSizeOfType(nullptr, tensor->type, &size_of_type) != kTfLiteOk) {
PyErr_SetString(PyExc_ValueError, "Unknown tensor type."); PyErr_SetString(PyExc_ValueError, "Unknown tensor type.");
free(data)
return nullptr; return nullptr;
} }
sparse_buffer_dims[0] = tensor->bytes / size_of_type; sparse_buffer_dims[0] = tensor->bytes / size_of_type;