code: def _add_identity_to_diagonal(x):
      return tf.linalg.set_diag(x, tf.linalg.diag_part(x) + 1.0)
  
imports: import collections
  import functools
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import mvn_tril
  from tensorflow_probability.python.internal import dtype_util
  from tensorflow_probability.python.internal import prefer_static
  from tensorflow_probability.python.internal import samplers
  from tensorflow_probability.python.math import linalg
  from tensorflow_probability.python.math.scan_associative import scan_associative
label_desc: Use image node_id 14 for calling a global function with example usage: _add_identity_to_diagonal(x) and returns: tf