Merge pull request #49775 from geetachavan1/cherrypicks_SHF4T

Prevent division by 0
tensorflow · May 30, 2021 · abebb8a · abebb8a
2 parents 199763a + aed3651
commit abebb8a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tensorflow/lite/kernels/embedding_lookup.cc b/tensorflow/lite/kernels/embedding_lookup.cc
@@ -68,6 +68,10 @@ TfLiteStatus EvalSimple(TfLiteContext* context, TfLiteNode* node,
                         const TfLiteTensor* lookup, const TfLiteTensor* value,
                         TfLiteTensor* output) {
   const int row_size = SizeOfDimension(value, 0);
+  if (row_size == 0) {
+    // Propagate empty tensor if input is empty
+    return kTfLiteOk;
+  }
   const int row_bytes = value->bytes / row_size;
 
   char* output_raw = GetTensorData<char>(output);