Different Behavior of tf.raw_ops.Cosh with jit_compile=True
Hi @zoux1a ,

This is related to #62212.

This is happening with complex inputs and outputs are generating Inf values which have a bug in assertion. The fix is proposed in eigen repo which is linked in above ticket.
 @zoux1a,
The above PR which was raised in the eigen repo was merged, So could you please try to use the latest tf-nightly version where the issue was resolved or not.
https://gitlab.com/libeigen/eigen/-/merge_requests/1431
 Thank you!
### Issue type

Bug

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

No

### Source

source

### TensorFlow version

2.14.0

### 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

GPU 0: NVIDIA GeForce RTX 2070 GPU 1: NVIDIA GeForce RTX 2070 GPU 2: NVIDIA GeForce RTX 2070 GPU 3: NVIDIA GeForce RTX 2070

### Current behavior?

When the tf.raw_ops.Cosh operation is invoked within a tf.function with JIT compilation enabled (jit_compile=True), it produces different results compared to the same operation called without JIT compilation. This inconsistency is observed when the code is executed on a GPU device.

### Standalone code to reproduce the issue

```shell
I can reproduce this issue on colab: https://colab.research.google.com/drive/1BvF1httSk5-QalMrtafvsj7QKKULwRa7?usp=sharing
```


### Relevant log output

```shell
File "/home/guihuan/LLM/results/tf-2/2023-10-22-20-21/test.py", line 28, in <module>
    tf.debugging.assert_near(no_op_res, op_res, atol=0.001, rtol=0.001)
  File "/home/guihuan/.conda/envs/night/lib/python3.9/site-packages/tensorflow/python/util/traceback_utils.py", line 153, in error_handler
    raise e.with_traceback(filtered_tb) from None
  File "/home/guihuan/.conda/envs/night/lib/python3.9/site-packages/tensorflow/python/ops/control_flow_assert.py", line 102, in Assert
    raise errors.InvalidArgumentError(
tensorflow.python.framework.errors_impl.InvalidArgumentError: Expected 'tf.Tensor(False, shape=(), dtype=bool)' to be true. Summarized data: b''
b'x and y not equal to tolerance rtol = tf.Tensor(0.001, shape=(), dtype=float64), atol = tf.Tensor(0.001, shape=(), dtype=float64)'
b'x (shape=(10, 9, 8) dtype=complex128) = '
(inf+0.7853981633974483j), (-inf+0.7853981633974483j), (-inf+0.7853981633974483j), ...
b'y (shape=(10, 9, 8) dtype=complex128) = '
(inf+0.7853981633974483j), (nan+nanj), (nan+nanj), ...
```
