TensorFlow Probability: Update References to `tfp.distributions`

Thank you for the suggestion! I'll make sure to update all references to use `tfp.distributions` instead of `tf.distributions`. This will ensure consistency and make it clear that we are referring to distributions from TensorFlow Probability.