def _sqrtx2p1(x):
sqrt_eps = np.sqrt(np.finfo(dtype_util.as_numpy_dtype(x.dtype)).eps)<EOL>return tf.where(<EOL>tf.abs(x) * sqrt_eps <= <NUM_LIT:1.>,<EOL>tf.sqrt(x**<NUM_LIT> + <NUM_LIT:1.>),<EOL>tf.abs(x))<EOL>
Implementation of `sqrt(1 + x**2)` which is stable despite large `x`.