`Overflow` in `tf.raw_ops.SparseReduceSumSparse` when there are too large values in `input_shape`. 
@xl1043237213,
Thank you for reporting the issue. I was able to reproduce the issue on tensorflow v2.15, v2.16 and tf-nightly. Kindly find the gist of it [here](https://colab.sandbox.google.com/gist/tilakrayal/608a78c05825ddeba08516a1893a5095/untitled1865.ipynb).
### Issue type

Bug

### Have you reproduced the bug with TensorFlow Nightly?

Yes

### Source

binary

### TensorFlow version

tf 2.15

### Custom code

No

### OS platform and distribution

Linux Ubuntu 20.04

### Mobile device

_No response_

### Python version

3.10

### Bazel version

_No response_

### GCC/compiler version

_No response_

### CUDA/cuDNN version

_No response_

### GPU model and memory

_No response_

### Current behavior?

`Overflow` in `tf.raw_ops.SparseReduceSumSparse` when there are too large values in `input_shape`.  See the colab link below for details.

### Standalone code to reproduce the issue

```shell
https://colab.research.google.com/drive/17mqn3t23p9sb4e3stOsPvZBTHENqJYku?usp=sharing
```


### Relevant log output

```shell
Non-OK-status: InitDims(dim_sizes) status: INVALID_ARGUMENT: Encountered overflow when multiplying 1700000000000 with 6000000, result: -8246744073709551616
```
