code: def _sqrt1px2(x):
      return tf.where(
          tf.math.abs(x)
          * np.sqrt(np.finfo(dtype_util.as_numpy_dtype(x.dtype)).eps)
          <= 1.0,
          tf.math.exp(0.5 * tf.math.log1p(tf.math.square(x))),
          tf.math.abs(x),
      )
  
imports: import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import custom_gradient
  from tensorflow_probability.python.internal import dtype_util
  from tensorflow_probability.python.internal import prefer_static
  from tensorflow_probability.python.math import generic
label_desc: Use image node_id 1 for calling a global function with example usage: _sqrt1px2(x) and returns: tf