code: def testSqrtWithFiniteGradsWithDynamicShape(self):
      x = tf1.placeholder_with_default([1.0], shape=[None])
      _, grad_tf_sqrt = value_and_gradient(tf.sqrt, x)
      _, grad_safe_sqrt = value_and_gradient(
          util.sqrt_with_finite_grads, x
      )
      self.assertAllEqual(
          *self.evaluate([grad_tf_sqrt, grad_safe_sqrt])
      )
  
imports: from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import tensorshape_util
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math.gradient import value_and_gradient
  from tensorflow_probability.python.math.psd_kernels.internal import util
label_desc: Use image node_id 11 for calling the UtilTest obj's underlying member method code with example usage: obj.testSqrtWithFiniteGradsWithDynamicShape() without return types