Merge pull request #49953 from tensorflow/cp2_1_369343977

Prevent yet another division by zero
tensorflow · Jun 2, 2021 · b4b2d71 · b4b2d71
2 parents 657e015 + 5995075
commit b4b2d71
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tensorflow/core/kernels/conv_grad_input_ops.cc b/tensorflow/core/kernels/conv_grad_input_ops.cc
@@ -673,6 +673,11 @@ class Conv2DCustomBackpropInputOp : public OpKernel {
         dims.batch_size == 1 ||
         thread_work_unit_size >= min_thread_work_unit_size;
 
+    OP_REQUIRES(
+        context, work_unit_size > 0,
+        errors::InvalidArgument("input, filter_sizes and out_backprop tensors "
+                                "must all have at least 1 element"));
+
     const size_t shard_size =
         use_parallel_contraction
             ? 1