def _upper_tail(self, tail_mass):
tail = helpers.estimate_tails(
self._logits_cumulative, tf.math.log(2 / tail_mass - 1),
tf.constant([self.batch_shape.num_elements(), 1, 1], tf.int32),
self.dtype)
return tf.reshape(tail, self.batch_shape_tensor())
def _upper_tail(self, tail_mass):
logits = -tf.math.log(tail_mass / 2 / (1. - tail_mass / 2))
return helpers.estimate_tails(self._logits_cumulative, logits,
self.batch_shape_tensor(), self.dtype)
def _quantization_offset(self):
return helpers.estimate_tails(
self._logits_cumulative, 0., self.batch_shape_tensor(), self.dtype)
