Skip to content

Commit

Permalink
Add deprecation notice to log_cumsum_exp.
Browse files Browse the repository at this point in the history
  • Loading branch information
emilyfertig committed Dec 6, 2022
1 parent 9600b80 commit 80e6f25
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions tensorflow_probability/python/math/generic.py
Expand Up @@ -30,6 +30,7 @@
from tensorflow_probability.python.internal import tensorshape_util
from tensorflow_probability.python.internal import variadic_reduce
from tensorflow_probability.python.math.scan_associative import scan_associative
from tensorflow.python.util import deprecation # pylint: disable=g-direct-tensorflow-import


__all__ = [
Expand Down Expand Up @@ -89,6 +90,9 @@ def log_combinations(n, counts, name='log_combinations'):

# TODO(b/154562929): Remove this once the built-in op supports XLA.
# TODO(b/156297366): Derivatives of this function may not always be correct.
@deprecation.deprecated('2023-03-01',
'`log_cumsum_exp` is deprecated; '
' Use `tf.math.cumulative_logsumexp` instead.')
def log_cumsum_exp(x, axis=-1, name=None):
"""Computes log(cumsum(exp(x))).
Expand Down

0 comments on commit 80e6f25

Please sign in to comment.