Skip to content

Commit 988a3f8

Browse files
committed
Apply _convert_to_tensor in exp implementation in NumPy/JAX backends.
Patched from d9deed0 .
1 parent 60edbd3 commit 988a3f8

File tree

1 file changed

+1
-1
lines changed
  • tensorflow_probability/python/internal/backend/numpy

1 file changed

+1
-1
lines changed

tensorflow_probability/python/internal/backend/numpy/numpy_math.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -568,7 +568,7 @@ def _divide_no_nan(x, y, name=None): # pylint: disable=unused-argument
568568

569569
exp = utils.copy_docstring(
570570
'tf.math.exp',
571-
lambda x, name=None: np.exp(x))
571+
lambda x, name=None: np.exp(_convert_to_tensor(x)))
572572

573573
expm1 = utils.copy_docstring(
574574
'tf.math.expm1',

0 commit comments

Comments
 (0)