Update deprecated calls of tfp.distributions.* to tfp.math*.
Update deprecated calls of tfp.distributions.* to tfp.math*.

PiperOrigin-RevId: 278049446

tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor