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

[LITE]Fix array leak during resize o/p

Browse Source
This commit is contained in:
Siju 2019-07-03 10:14:29 +05:30 committed by GitHub
parent e0ab930a7e
commit 57f1af3b8b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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/kernels/fill.cc
View File

@ -38,6 +38,7 @@ TfLiteStatus ResizeOutputImpl(TfLiteContext* context, const TfLiteTensor* dims,
for (int i = 0; i < output_shape->size; ++i) { for (int i = 0; i < output_shape->size; ++i) {
T data = GetTensorData<T>(dims)[i]; T data = GetTensorData<T>(dims)[i];
if (data < 0) { if (data < 0) {
TfLiteIntArrayFree(output_shape);
context->ReportError(context, "Fill dimensions must be >= 0", dims->type); context->ReportError(context, "Fill dimensions must be >= 0", dims->type);
return kTfLiteError; return kTfLiteError;
} }