tf.raw_ops.QuantizeAndDequantizeV3: Aborted (core dumped)
@nonenone12135,
I was able to reproduce the issue on both tensorflow v2.15 and tf-nightly. Kindly find the gist of it [here](https://colab.research.google.com/gist/tilakrayal/ddb3681c3a808fbd03695d8bde5bdc38/untitled1771.ipynb). Please allow some time to deep dive into the issue and provide the resolution. Thank you!
### Issue type

Bug

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

Yes

### Source

binary

### TensorFlow version

tf 2.15

### Custom code

Yes

### OS platform and distribution

Ubuntu 20.04

### Mobile device

_No response_

### Python version

3.9

### 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.QuantizeAndDequantizeV3` encounters "Aborted (core dumped)".

### Standalone code to reproduce the issue

```shell
import tensorflow as tf

input_data = tf.constant([1.5, 2.5, 3.5, 4.5])
input_min = tf.constant([[[0.0, 0.0], [0.0, 0.0]], [[0.0, 0.0], [0.0, 0.0]], [[0.0, 0.0], [0.0, 0.0]]])

input_max = tf.constant(5.0)
num_bits = tf.constant(8)
signed_input = True
range_given = True
narrow_range = False
axis = -1

output_data = tf.raw_ops.QuantizeAndDequantizeV3(input=input_data, input_min=input_min, input_max=input_max, num_bits=num_bits, signed_input=signed_input, range_given=range_given, narrow_range=narrow_range, axis=axis)

print(output_data)
```


### Relevant log output

```shell
2024-03-14 05:50:03.159345: F tensorflow/core/framework/tensor.cc:852] Check failed: 1 == NumElements() (1 vs. 2)Must have a one element tensor
Aborted (core dumped)
```
