Aborted (core dumped) in `tf.raw_ops.ResourceApplyAdagrad/tf.raw_ops.ResourceApplyAdagradDA/tf.raw_ops.ResourceApplyAdagradV2`
Hi @x0w3n ,

I was able to reproduce the issue on Colab using TF [v2.16](https://colab.research.google.com/gist/sushreebarsa/5cec0f7080f1672e116f91f6f9cc72a2/69285.ipynb
) and TF-[nightly](https://colab.research.google.com/gist/sushreebarsa/1aa2c02a796f7ca238a92fa601e8d0e7/69285.ipynb#scrollTo=M3Qm-JLCymRs) ,Please find the gist there for reference.
Thank you!
### 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?

On specific input, tf.raw_ops.ResourceApplyAdagrad/tf.raw_ops.ResourceApplyAdagradDA/tf.raw_ops.ResourceApplyAdagradV2 triggers "Aborted (core dumped)". We ran the code on colab's latest TensorFlow Nightly, which also triggers the crash. The cause of the crash may be the grad parameter.

### Standalone code to reproduce the issue

```shell
import tensorflow as tf

var = tf.Variable([1.0, 2.0, 3.0])
accum = tf.Variable([0.1, 0.2, 0.3], dtype=tf.complex64) 
accum2 = tf.Variable([0, 1.2, -3.3], dtype=tf.complex64) 
lr = 0.01
grad = tf.constant([0.1, 0.2, 0.3])
momentum = 0.9

# crash
tf.raw_ops.ResourceApplyAdagrad(
    var=var.handle,
    accum=accum.handle,
    lr=lr,
    grad=grad,
    use_locking=False,
)

# crash
# tf.raw_ops.ResourceApplyAdagradDA(var=var.handle,
#               gradient_accumulator=accum.handle,
#               gradient_squared_accumulator=accum2.handle,
#               grad=grad,  lr=lr,
#               l1=lr, l2=lr, global_step=1000,
#               use_locking=False)

# crash
# tf.raw_ops.ResourceApplyAdagradV2(var=var.handle,
#                 accum=accum.handle,
#                 epsilon = 0.9,
#                 grad=grad, lr=lr,
#                 use_locking=False,update_slots=False)
```


### Relevant log output

```shell
2024-06-06 01:18:44.797518: I tensorflow/core/util/port.cc:113] oneDNN custom operations are on. You may see slightly different numerical results due to floating-point round-off errors from different computation orders. To turn them off, set the environment variable `TF_ENABLE_ONEDNN_OPTS=0`.
2024-06-06 01:18:44.829812: 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 AVX512F AVX512_VNNI FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags.
2024-06-06 01:18:45.875807: F tensorflow/core/framework/tensor.cc:844] Check failed: dtype() == expected_dtype (8 vs. 1) float expected, got complex64
Aborted (core dumped)
```
