OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-07 12:20:24 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Use TF_LITE_KERNEL_LOG instead of report_error

Browse Source 
PiperOrigin-RevId: 383535580
Change-Id: I83186751d13dcb27515831743563f61d0ff2fdf2
This commit is contained in:
Jacques Pienaar 2021-07-07 18:33:48 -07:00 committed by TensorFlower Gardener
parent 985de0ce26
commit 55acb46ee7
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/lite/kernels/split_v.cc
View File

@ -76,7 +76,7 @@ TfLiteStatus ResizeOutputTensors(TfLiteContext* context, TfLiteNode* node,
} else if (size_splits->type == kTfLiteInt64) {
GetSizeSplitsVector<int64_t>(size_splits, &size_splits_vector);
} else {
context->ReportError(context, "size_splits only support type int32|int64.");
TF_LITE_KERNEL_LOG(context, "size_splits only support type int32|int64.");
return kTfLiteError;
}
@ -88,8 +88,9 @@ TfLiteStatus ResizeOutputTensors(TfLiteContext* context, TfLiteNode* node,
if (minus_one_index == -1) {
minus_one_index = i;
} else {
context->ReportError(context,
TF_LITE_KERNEL_LOG(context,
"The size_splits contains more than one -1.");
return kTfLiteError;
}
} else {
size_splits_sum += size_splits_vector.at(i);