def cond_top_dist(self, cond_latents):
    """Get a conditional prior distribution on the top latent."""
    prior_dist = self.top_cond_prior("conditional", cond_latents)
    return prior_dist.loc, prior_dist.scale
Get a conditional prior distribution on the top latent.
tensor2tensor/models/video/next_frame_glow.py