Merge pull request #49746 from geetachavan1/cherrypicks_UEIS8

Fix another division by 0 in TFLite
tensorflow · May 26, 2021 · 8c0063e · 8c0063e
2 parents 558af1d + 643a8b9
commit 8c0063e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tensorflow/lite/kernels/transpose_conv.cc b/tensorflow/lite/kernels/transpose_conv.cc
@@ -492,6 +492,10 @@ TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
   const auto* params =
       reinterpret_cast<TfLiteTransposeConvParams*>(node->builtin_data);
 
+  // Prevent divisions by 0
+  TF_LITE_ENSURE(context, params->stride_height > 0);
+  TF_LITE_ENSURE(context, params->stride_width > 0);
+
   // Resize any deferred dynamic tensors
   if (IsDynamicTensor(output)) {
     TF_LITE_ENSURE_OK(context, ResizeTensor(context, output_shape, output));