From 87fc8379200e1faf536d74281c062820e33be75e Mon Sep 17 00:00:00 2001
From: Gaurav Singh <gaurav1086@gmail.com>
Date: Sat, 2 May 2020 14:50:25 -0400
Subject: [PATCH] [Lite] data: Fix memory leak

---
 .../lite/python/interpreter_wrapper/interpreter_wrapper.cc       | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tensorflow/lite/python/interpreter_wrapper/interpreter_wrapper.cc b/tensorflow/lite/python/interpreter_wrapper/interpreter_wrapper.cc
index bd78d56172e..313de20595d 100644
--- 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;