Use `tf.gradient` in graph mode in `tfp.math.value_and_gradient`.
Use `tf.gradient` in graph mode in `tfp.math.value_and_gradient`.

PiperOrigin-RevId: 234193435

tensorflow/probability,tensorflow/probability