Aborted (core dumped) in `tf.scatter_nd`
Hi @x0w3n ,
- I was able to reproduce the issue on Colab using TF v2.15 and TF-nightly ,Please find the [gist](https://colab.research.google.com/gist/Venkat6871/140c866714fb864890903cb3a5380a36/69221_2-15-nightly-v.ipynb) here for reference.

Thank you! Thank you for your response. Are you satisfied with the resolution of your issue?
<a href="https://docs.google.com/forms/d/e/1FAIpQLSfaP12TRhd9xSxjXZjcZFNXPGk4kc1-qMdv3gc6bEP90vY1ew/viewform?entry.85265664=Yes&entry.2137816233=https://github.com/tensorflow/tensorflow/issues/69221">Yes</a>
<a href="https://docs.google.com/forms/d/e/1FAIpQLSfaP12TRhd9xSxjXZjcZFNXPGk4kc1-qMdv3gc6bEP90vY1ew/viewform?entry.85265664=No&entry.2137816233=https://github.com/tensorflow/tensorflow/issues/69221">No</a>

### Issue type

Bug

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

Yes

### Source

source

### TensorFlow version

tf 2.16

### Custom code

Yes

### OS platform and distribution

Linux Ubuntu 22.04.3 LTS (x86_64)

### Mobile device

_No response_

### Python version

3.9.13

### Bazel version

_No response_

### GCC/compiler version

_No response_

### CUDA/cuDNN version

_No response_

### GPU model and memory

_No response_

### Current behavior?

Triggers crash when setting boundary values for the indices parameter of tf.scatter_nd.

### Standalone code to reproduce the issue

```shell
import tensorflow as tf

indices = tf.constant([[4], [3], [1], [7]], dtype=tf.int32)
updates = tf.constant([9, 10, 11, 12], dtype=tf.int32)
shape = tf.cast([8], dtype=tf.int32)
scatter_nd = tf.scatter_nd(tf.expand_dims(indices, 1), updates, shape)
```


### Relevant log output

```shell
2024-06-05 10:47:19.247456: I tensorflow/core/platform/cpu_feature_guard.cc:210] This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations.
To enable the following instructions: AVX2 FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags.
2024-06-05 10:47:20.277026: F tensorflow/core/framework/tensor_shape.cc:357] Check failed: d < dims() (1 vs. 1)
Aborted (core dumped)
```
