[Lite] data: Fix memory leak

2020-05-02 14:50:25 -04:00 · 2020-05-02 14:50:25 -04:00 · 87fc837920
commit 87fc837920
parent 25b0e7d995
1 changed files with 1 additions and 0 deletions
--- a/tensorflow/lite/python/interpreter_wrapper/interpreter_wrapper.cc
+++ b/tensorflow/lite/python/interpreter_wrapper/interpreter_wrapper.cc
@ -592,6 +592,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;