tf.random.truncated_normal (inside graph) crashed TPUv4 pod
@edwardyehuang 
Could you please provide reproducible code or Colab gist to replicate the issue reported here ?

Thank you ! This issue is stale because it has been open for 7 days with no activity. It will be closed if no further activity occurs. Thank you. This issue was closed because it has been inactive for 7 days since being marked as stale. Please reopen if you'd like to work on this further. 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/60291">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/60291">No</a>

 ### Issue Type

Bug

### Source

source

### Tensorflow Version

tf 2.10 TPU Pod

### Current Behaviour?


I found ```tf.random.truncated_normal``` will crash the TPUv4 Pod during training (graph computation).

```tf.random.truncated_normal``` is fine when not in a graph (e.g. variable initialization).

```tf.random.normal``` and ```tf.random.uniform``` are fine during training.


### Relevant log output
