Fix build errors

This commit is contained in:
Gaurav Singh 2020-05-06 09:25:09 -04:00
parent 87fc837920
commit c7828e73f7

View File

@ -592,7 +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)
free(data);
return nullptr;
}
sparse_buffer_dims[0] = tensor->bytes / size_of_type;