core dumped with tf.raw_ops.BiasAdd
@sachinprasadhs,
I was able to reproduce the issue on tensorflow v2.14 and [tf-nightly](https://colab.research.google.com/gist/tilakrayal/627c4a38027beada5fb0a4da7bc8e915/untitled1396.ipynb) But whereas in tensorflow v2.13, the colab was unable to get a crash and it was executed with the error. Kindly find the gist of it [here](https://colab.research.google.com/gist/tilakrayal/f402321fdfdb57589971f9c77e499bb4/untitled1397.ipynb). 
### Issue type

Bug

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

Yes

### Source

source

### TensorFlow version

v1.12.1-100714-gd8e55c05473 2.15.0-dev20231005

### Custom code

Yes

### OS platform and distribution

Ubuntu 22.04.3 LTS (x86_64)

### Mobile device

_No response_

### Python version

_No response_

### Bazel version

_No response_

### GCC/compiler version

_No response_

### CUDA/cuDNN version

11.8

### GPU model and memory

_No response_

### Current behavior?

core dumped when running tf.raw_ops.BiasAdd with below code.



### Standalone code to reproduce the issue

```shell
import tensorflow as tf 
args = {'value':  tf.random.uniform([8, 2], 0, 256, dtype=tf.int32), 'bias':  tf.random.uniform([5],0, 256, dtype=tf.int32), 'data_format': 'NCHW'}
res = tf.raw_ops.BiasAdd(**args)
print(res)
```


### Relevant log output

```shell
2023-10-18 13:29:08.644285: F tensorflow/core/framework/tensor_shape.cc:357] Check failed: d < dims() (2 vs. 2)
Aborted (core dumped)
```
