diff --git a/tensorflow_probability/python/internal/backend/numpy/ops.py b/tensorflow_probability/python/internal/backend/numpy/ops.py index 2d396f5ef4..b96b22128b 100644 --- a/tensorflow_probability/python/internal/backend/numpy/ops.py +++ b/tensorflow_probability/python/internal/backend/numpy/ops.py @@ -394,7 +394,8 @@ def __init__(self, *args, **kwargs): broadcast_to = utils.copy_docstring( 'tf.broadcast_to', - lambda input, shape, name=None: np.broadcast_to(input, shape)) + lambda input, shape, name=None: np.broadcast_to( + _convert_to_tensor(input), shape)) def _cast(x, dtype):