Merge pull request #53998 from pranve/cherrypick-fcd18ce3101f245b083b…

…30655c27b239dc72221e-on-r2.7

Prevent integer overflow in `OpLevelCostEstimator::CalculateTensorSize`.

tensorflow/core/grappler/costs/op_level_cost_estimator.cc

@@ -1554,7 +1554,13 @@ int64_t OpLevelCostEstimator::CalculateTensorSize(
   int64_t count = CalculateTensorElementCount(tensor, found_unknown_shapes);
   int size = DataTypeSize(BaseType(tensor.dtype()));
   VLOG(2) << "Count: " << count << " DataTypeSize: " << size;
-  return count * size;
+  int64_t tensor_size = MultiplyWithoutOverflow(count, size);
+  if (tensor_size < 0) {
+    VLOG(1) << "Overflow encountered when computing tensor size, multiplying "
+            << count << " with " << size;
+    return -1;
+  }
+  return tensor_size;
 }
 
 int64_t OpLevelCostEstimator::CalculateInputSize(const OpInfo& op_info,