Fix overflow CHECK issue with tf.raw_ops.AddManySparseToTensorsMap.

PiperOrigin-RevId: 369572755
Change-Id: Ib054995f5564a15304772c77a27fbef93c7ce9a7

Author: tensorflower-gardener

tensorflow/core/kernels/sparse_concat_op.cc

@@ -27,7 +27,6 @@ limitations under the License.
 #include "tensorflow/core/framework/tensor_util.h"
 #include "tensorflow/core/framework/types.h"
 #include "tensorflow/core/lib/gtl/inlined_vector.h"
-#include "tensorflow/core/util/overflow.h"
 #include "tensorflow/core/util/sparse/sparse_tensor.h"
 
 namespace tensorflow {
@@ -68,25 +67,10 @@ class SparseConcatOp : public OpKernel {
                 errors::InvalidArgument("Expected ", N, " input shapes, got ",
                                         shapes.size()));
     for (int i = 0; i < N; i++) {
-      int new_num_elements = 1;
-      bool overflow_ocurred = false;
       OP_REQUIRES(context, TensorShapeUtils::IsVector(shapes[i].shape()),
                   errors::InvalidArgument(
                       "Input shapes should be a vector but received shape ",
                       shapes[i].shape().DebugString(), " at position ", i));
-      auto input_shape_vector = shapes[i].vec<int64>();
-      for (int j = 0; j < input_shape_vector.size(); j++) {
-        new_num_elements =
-            MultiplyWithoutOverflow(new_num_elements, input_shape_vector(j));
-        if (new_num_elements < 0) {
-          overflow_ocurred = true;
-          break;
-        }
-      }
-
-      OP_REQUIRES(
-          context, !overflow_ocurred,
-          errors::Internal("Encountered overflow from large input shape."));
     }
 
     const TensorShape input_shape(shapes[0].vec<int64>());