add missing convert_to_tensor (#19578)

Author: lkarthee

keras/src/backend/mlx/math.py

@@ -31,6 +31,7 @@ def in_top_k(targets, predictions, k):
 
 
 def logsumexp(x, axis=None, keepdims=False):
+    x = convert_to_tensor(x)
     return mx.logsumexp(x, axis, keepdims)
 
 
@@ -87,10 +88,12 @@ def istft(
 
 
 def rsqrt(x):
+    x = convert_to_tensor(x)
     return mx.rsqrt(x)
 
 
 def erf(x):
+    x = convert_to_tensor(x)
     return mx.erf(x)
 
 

keras/src/backend/mlx/nn.py

@@ -308,6 +308,8 @@ def moments(x, axes, keepdims=False, synchronized=False):
 def batch_normalization(
     x, mean, variance, axis, offset=None, scale=None, epsilon=1e-3
 ):
+    x = convert_to_tensor(x)
+    mean = convert_to_tensor(x)
     shape = [1] * len(x.shape)
     shape[axis] = mean.shape[0]
     mean = mx.reshape(mean, shape)

keras/src/backend/mlx/numpy.py

@@ -82,6 +82,7 @@ def absolute(x):
 
 
 def abs(x):
+    x = convert_to_tensor(x)
     return absolute(x)
 
 
@@ -164,10 +165,12 @@ def argmax(x, axis=None, keepdims=False):
 
 
 def argmin(x, axis=None, keepdims=False):
+    x = convert_to_tensor(x)
     return mx.argmin(x, axis=axis, keepdims=keepdims)
 
 
 def argsort(x, axis=-1):
+    x = convert_to_tensor(x)
     return mx.argsort(x, axis=axis)
 
 
@@ -235,6 +238,7 @@ def conjugate(x):
 
 
 def conj(x):
+    x = convert_to_tensor(x)
     return conjugate(x)
 
 
@@ -916,6 +920,8 @@ def divide_no_nan(x1, x2):
 
 
 def true_divide(x1, x2):
+    x1 = convert_to_tensor(x1)
+    x2 = convert_to_tensor(x2)
     return divide(x1, x2)