message: TFLite: Enhance input check for ResizeNearestNeghbor
diff: @@ -57,11 +57,13 @@ TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {
  const TfLiteTensor* size = GetInput(context, node, kSizeTensor);
  TfLiteTensor* output = GetOutput(context, node, kOutputTensor);
  - // TODO(ahentz): Our current implementations rely on the inputs being 4D.
  + // TODO(ahentz): Our current implementations rely on the input being 4D,
  + // and the size being 1D tensor with exactly 2 elements.
  TF_LITE_ENSURE_EQ(context, NumDimensions(input), 4);
  TF_LITE_ENSURE_EQ(context, NumDimensions(size), 1);
  -
  TF_LITE_ENSURE_EQ(context, size->type, kTfLiteInt32);
  + TF_LITE_ENSURE_EQ(context, size->dims->data[0], 2);
  +
  output->type = input->type;
  if (!IsConstantTensor(size)) {
  