description: Augments doc on tfd.kl_divergence, EachDist.kl_divergence/cross_entropy.
View source on GitHub |
Augments doc on tfd.kl_divergence, EachDist.kl_divergence/cross_entropy.
tfp.layers.dense_variational_v2.kullback_leibler.augment_kl_xent_docs(
distributions_module
)