Merge pull request #49446 from geetachavan1/cherrypicks_78PN8

Prevent division by 0 in `QuantizedBiasAdd`.
tensorflow · May 21, 2021 · ece05f0 · ece05f0
2 parents 7e61c54 + 4d32c32
commit ece05f0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tensorflow/core/kernels/quantized_bias_add_op.cc b/tensorflow/core/kernels/quantized_bias_add_op.cc
@@ -56,6 +56,8 @@ class QuantizedBiasAddOp : public OpKernel {
             "Must provide as many biases as the last dimension "
             "of the input tensor: ",
             bias.shape().DebugString(), " vs. ", input.shape().DebugString()));
+    OP_REQUIRES(context, bias.NumElements() > 0,
+                errors::InvalidArgument("Must provide at least 1 bias"));
 
     Tensor* output = nullptr;
     OP_REQUIRES_OK(context,