From 646754898f631336a135e85220f99185b3d09468 Mon Sep 17 00:00:00 2001
From: jburnim <jburnim@google.com>
Date: Fri, 16 Feb 2024 13:59:20 -0800
Subject: [PATCH] Add missing convert_to_tensor in broadcast_to for NumPy/JAX
 backends.

PiperOrigin-RevId: 607793016
---
 tensorflow_probability/python/internal/backend/numpy/ops.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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):