Searched refs:_distribution (Results 1 – 10 of 10) sorted by relevance
29 self._distribution = distribution33 return self._distribution96 with self._distribution.scope():
108 self._distribution = distribution112 if self._distribution:113 self._scope = self._distribution.scope()117 if self._distribution:
131 self._distribution = distribution146 dtype=self._distribution.dtype,147 reparameterization_type=self._distribution.reparameterization_type,149 allow_nan_stats=self._distribution.allow_nan_stats,160 return self._distribution
300 self._distribution = poisson_lib.Poisson(331 return self._distribution
125 self._distribution = distribution141 return self._distribution
490 self._distribution = distribution539 return self._distribution
315 self._distribution = distribution318 dtype=self._distribution.dtype,319 reparameterization_type=self._distribution.reparameterization_type,321 allow_nan_stats=self._distribution.allow_nan_stats,332 return self._distribution
79 self.assertAllClose(logits, dist._distribution.logits.eval())80 self.assertAllEqual([3], dist._distribution.logits.get_shape())
284 def _distribution(self, state): member in _SwapRegretOptimizer368 distribution = self._distribution(state)
246 ret._distribution = distribution449 with self._distribution.extended.colocate_vars_with(colocate_with):