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

Merge pull request #39104 from gaurav1086:intepreter_memory_leak

Browse Source 
PiperOrigin-RevId: 311266368
Change-Id: I817152b28d3ab9b04c34e71dde1ac91dc5b90a32
This commit is contained in:
TensorFlower Gardener 2020-05-12 22:21:27 -07:00
commit 75507c75e8
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

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