aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/ops/distributions/bernoulli.py
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/python/ops/distributions/bernoulli.py')
-rw-r--r--tensorflow/python/ops/distributions/bernoulli.py20
1 files changed, 0 insertions, 20 deletions
diff --git a/tensorflow/python/ops/distributions/bernoulli.py b/tensorflow/python/ops/distributions/bernoulli.py
index 4c16d62e9a..68aaf3815e 100644
--- a/tensorflow/python/ops/distributions/bernoulli.py
+++ b/tensorflow/python/ops/distributions/bernoulli.py
@@ -157,26 +157,6 @@ class Bernoulli(distribution.Distribution):
return math_ops.cast(self.probs > 0.5, self.dtype)
-class BernoulliWithSigmoidProbs(Bernoulli):
- """Bernoulli with `probs = nn.sigmoid(logits)`."""
-
- def __init__(self,
- logits=None,
- dtype=dtypes.int32,
- validate_args=False,
- allow_nan_stats=True,
- name="BernoulliWithSigmoidProbs"):
- parameters = locals()
- with ops.name_scope(name):
- super(BernoulliWithSigmoidProbs, self).__init__(
- probs=nn.sigmoid(logits, name="sigmoid_probs"),
- dtype=dtype,
- validate_args=validate_args,
- allow_nan_stats=allow_nan_stats,
- name=name)
- self._parameters = parameters
-
-
@kullback_leibler.RegisterKL(Bernoulli, Bernoulli)
def _kl_bernoulli_bernoulli(a, b, name=None):
"""Calculate the batched KL divergence KL(a || b) with a and b Bernoulli.