From c66fd8abdce5e7243c8d157a43ef9ac905a8fef1 Mon Sep 17 00:00:00 2001 From: Esun Kim Date: Fri, 30 Jan 2026 22:54:26 -0800 Subject: [PATCH] Missing kTfLiteInt16 in ComparisonsPrepare --- tensorflow/lite/micro/kernels/comparisons.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/lite/micro/kernels/comparisons.cc b/tensorflow/lite/micro/kernels/comparisons.cc index ca308a59e9d..ea9778cd72e 100644 --- a/tensorflow/lite/micro/kernels/comparisons.cc +++ b/tensorflow/lite/micro/kernels/comparisons.cc @@ -559,7 +559,7 @@ TfLiteStatus ComparisonsPrepare(TfLiteContext* context, TfLiteNode* node) { micro_context->AllocateTempInputTensor(node, kInputTensor2); TF_LITE_ENSURE(context, input2 != nullptr); - if (input1->type == kTfLiteInt8) { + if (input1->type == kTfLiteInt8 || input1->type == kTfLiteInt16) { auto input1_offset = -input1->params.zero_point; auto input2_offset = -input2->params.zero_point; const int kLeftShift = 8;