Searched refs:distribution_fn (Results 1 – 2 of 2) sorted by relevance
121 distribution_fn, argument159 self._distribution_fn = distribution_fn161 self._distribution0 = (distribution_fn() if sample0 is None162 else distribution_fn(sample0))181 def distribution_fn(self): member in Autoregressive215 samples = self.distribution_fn(samples).sample(seed=seed)219 return self.distribution_fn(value).log_prob(value)222 return self.distribution_fn(value).prob(value)
307 def __init__(self, name, distribution_fn, required_gpus=None, argument309 self._distribution_fn = distribution_fn