commit 7f283ff806b2031f407db64c4d3edcda8fb9f9f5
author Amit Patankar <amitpatankar@google.com> Wed Apr 28 15:00:39 2021 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Wed Apr 28 15:08:18 2021 -0700
tree abed11d6dab3862431b5179e4275772310737e6b
parent 05a63e605a31e86c5dd96c5c8a763eda9ac7bb33

Fix FPE issue in external Eigen source code issue with `tf.raw_ops.SparseMatMul`.

PiperOrigin-RevId: 370992919
Change-Id: Icfb276fef5fb40928b27c3e44608d2aca72c9fd7

tensorflow/core/kernels/sparse_matmul_op.cc

diff --git a/tensorflow/core/kernels/sparse_matmul_op.cc b/tensorflow/core/kernels/sparse_matmul_op.cc
index f574785..a02afaf 100644
--- a/tensorflow/core/kernels/sparse_matmul_op.cc
+++ b/tensorflow/core/kernels/sparse_matmul_op.cc
@@ -1039,6 +1039,10 @@
     if (transpose_b) {
       // TODO(agarwal): avoid transposing the matrix here and directly handle
       // transpose in CreateDenseSlices.
+      OP_REQUIRES(ctx, right->dim_size(0) != 0,
+                  errors::InvalidArgument("b has an entry 0 in it's shape."));
+      OP_REQUIRES(ctx, right->dim_size(1) != 0,
+                  errors::InvalidArgument("b has an entry 0 in it's shape."));
       right_tr.reset(
           new Tensor(right->dtype(),
                      TensorShape({right->dim_size(1), right->dim_size(0)})));