kullback - Robuta Search

https://www.tensorflow.org/probability/api_docs/python/tfp/layers/dense_variational_v2/kullback_leibler/augment_kl_xent_docs

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