A check fail can be triggered in TridiagonalSolve
@sachinprasadhs I was able to replicate the issue on colab using TF v[2.12](https://colab.research.google.com/gist/sushreebarsa/23002b3c2d4cbd311f739dcdd37e2157/60198.ipynb) and tf-[nightly](https://colab.research.google.com/gist/sushreebarsa/fa9c18ffb6f4829647d36d78942c05af/untitled784.ipynb#scrollTo=hIczSZUMnMVl)(2.13.0-dev20230403), please find the attached gists. Thank you! 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/60198">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/60198">No</a>

<details><summary>Click to expand!</summary> 
 
 ### Issue Type

Bug

### Have you reproduced the bug with TF nightly?

Yes

### Source

binary

### Tensorflow Version

2.13.0-dev20230331

### Custom Code

No

### OS Platform and Distribution

Linux Ubuntu 20.04

### Mobile device

_No response_

### Python version

3.10

### Bazel version

_No response_

### GCC/Compiler version

_No response_

### CUDA/cuDNN version

CUDA 11.8

### GPU model and memory

_No response_

### Current Behaviour?

```shell
The following code can trigger a crash in `tf.raw_ops.TridiagonalSolve` due to check-fail in the latest version of TensorFlow.
```


### Standalone code to reproduce the issue

```shell
import tensorflow as tf
with tf.device("GPU:0"):
    partial_pivoting = True
    perturb_singular = False
    diagonals = tf.complex(tf.random.uniform([2, 13, 0, 13, 857005098598382018], dtype=tf.float32, minval=-1024, maxval=1024),tf.random.uniform([2, 13, 0, 13, 857005098598382018], dtype=tf.float32, minval=-1024, maxval=1024))
    rhs = tf.complex(tf.random.uniform([15, 15, 12, 1, 10, 16], dtype=tf.float32, minval=-1024, maxval=1024),tf.random.uniform([15, 15, 12, 1, 10, 16], dtype=tf.float32, minval=-1024, maxval=1024))
    res = tf.raw_ops.TridiagonalSolve(
        partial_pivoting=partial_pivoting,
        perturb_singular=perturb_singular,
        diagonals=diagonals,
        rhs=rhs,
    )
```


### Relevant log output

```shell
2023-04-01 16:22:38.864472: I tensorflow/core/util/port.cc:110] 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`.
2023-04-01 16:22:38.915491: I tensorflow/core/platform/cpu_feature_guard.cc:182] 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.
2023-04-01 16:22:39.745143: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Could not find TensorRT
2023-04-01 16:22:41.321428: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1635] Created device /job:localhost/replica:0/task:0/device:GPU:0 with 14561 MB memory:  -> device: 0, name: Tesla V100-PCIE-16GB, pci bus id: 0000:2f:00.0, compute capability: 7.0
2023-04-01 16:22:41.585403: F tensorflow/core/framework/tensor_shape.cc:201] Non-OK-status: InitDims(dim_sizes) status: INVALID_ARGUMENT: Encountered overflow when multiplying 13 with 857005098598382018, result: -7305677791930585382
Aborted (core dumped)
```
</details>