Save the date! Google I/O returns May 18-20 Register now

tfp.layers.dense_variational_v2.kullback_leibler.augment_kl_xent_docs

Augments doc on tfd.kl_divergence, EachDist.kl_divergence/cross_entropy.