Add int16 as possible output type to the error message

2020-07-17 12:53:59 +01:00 · 2020-07-17 12:53:59 +01:00 · f6c0221749
commit f6c0221749
parent 860898de6f
1 changed files with 1 additions and 1 deletions
--- a/tensorflow/lite/kernels/resize_nearest_neighbor.cc
+++ b/tensorflow/lite/kernels/resize_nearest_neighbor.cc
@ -128,7 +128,7 @@ TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
        GetTensorShape(output), GetTensorData<int16_t>(output));
  } else {
    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));
    return kTfLiteError;
  }