Merge pull request #49709 from geetachavan1/cherrypicks_FR9R6

Fix division by zero in TFLite padding.
tensorflow · May 26, 2021 · 161f244 · 161f244
2 parents 711324e + 3c40555
commit 161f244
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tensorflow/lite/kernels/padding.h b/tensorflow/lite/kernels/padding.h
@@ -44,6 +44,11 @@ inline int ComputePaddingWithOffset(int stride, int dilation_rate, int in_size,
 inline int ComputeOutSize(TfLitePadding padding, int image_size,
                           int filter_size, int stride, int dilation_rate = 1) {
   int effective_filter_size = (filter_size - 1) * dilation_rate + 1;
+
+  // TODO(b/186448822): This uses 0 since the function has no other way to
+  // report error case
+  if (stride == 0) return 0;
+
   switch (padding) {
     case kTfLitePaddingSame:
       return (image_size + stride - 1) / stride;