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

Add int16 as possible output type to the error message

Browse Source
This commit is contained in:
Thibaut Goetghebuer-Planchon 2020-07-17 12:53:59 +01:00
parent 860898de6f
commit f6c0221749
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/lite/kernels/resize_nearest_neighbor.cc
View File

@ -128,7 +128,7 @@ TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
GetTensorShape(output), GetTensorData<int16_t>(output)); GetTensorShape(output), GetTensorData<int16_t>(output));
} else { } else {
TF_LITE_KERNEL_LOG(context, TF_LITE_KERNEL_LOG(context,
"Output type is %s, requires float, uint8 or int8.", "Output type is %s, requires float, uint8, int8 or int16.",
TfLiteTypeGetName(output->type)); TfLiteTypeGetName(output->type));
return kTfLiteError; return kTfLiteError;
} }