Aborted (core dumped) with `tf.raw_ops.LoadAndRemapMatrix`
Hi **@LongZE666** ,
I tried to run your code on Colab using TF v2.16.1, nightly and faced the same issue. Please find the [gist](https://colab.sandbox.google.com/gist/Venkat6871/c35da15e8fe8a8692af9c2ad8bd2d9db/64655-2-16-1-nightly.ipynb) here for reference.
Thank you!
### Issue type

Bug

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

Yes

### Source

source

### TensorFlow version

tf 2.16.1

### Custom code

Yes

### OS platform and distribution

Ubuntu 20.04

### Mobile device

_No response_

### Python version

3.11

### Bazel version

_No response_

### GCC/compiler version

_No response_

### CUDA/cuDNN version

_No response_

### GPU model and memory

_No response_

### Current behavior?

Under specific input, `tf.raw_ops.tf.raw_ops.LoadAndRemapMatrix` encounters "Aborted (core dumped)".

### Standalone code to reproduce the issue

```shell
import tensorflow as tf

num_rows = 7
num_cols = 16
max_rows_in_memory = -1
ckpt_path = tf.constant("/tmp", shape=[1], dtype=tf.string)
old_tensor_name = tf.constant("/tmp", shape=[1], dtype=tf.string)
row_remapping = tf.constant(-1, shape=[7], dtype=tf.int64)
col_remapping = tf.constant(0, shape=[], dtype=tf.int64)
initializing_values = tf.constant(42, shape=[112], dtype=tf.float32)
tf.raw_ops.LoadAndRemapMatrix(ckpt_path=ckpt_path, old_tensor_name=old_tensor_name, row_remapping=row_remapping, col_remapping=col_remapping, initializing_values=initializing_values, num_rows=num_rows, num_cols=num_cols, max_rows_in_memory=max_rows_in_memory)
```


### Relevant log output

```shell
2024-03-28 07:29:28.849010: F tensorflow/core/framework/tensor_shape.cc:45] Check failed: NDIMS == dims() (1 vs. 0)Asking for tensor of 1 dimensions from a tensor of 0 dimensions
Aborted (core dumped)
```
