diff --git a/keras/src/backend/tensorflow/numpy.py b/keras/src/backend/tensorflow/numpy.py
index b1f9ef211618..7468ca8db92b 100644
--- a/keras/src/backend/tensorflow/numpy.py
+++ b/keras/src/backend/tensorflow/numpy.py
@@ -2134,7 +2134,7 @@ def searchsorted(sorted_sequence, values, side="left"):
             f"sorted_sequence.shape={sorted_sequence.shape}"
         )
     out_type = (
-        "int32" if len(sorted_sequence) <= np.iinfo(np.int32).max else "int64"
+        "int32" if sorted_sequence.shape[0] <= np.iinfo(np.int32).max else "int64"
     )
     return tf.searchsorted(
         sorted_sequence, values, side=side, out_type=out_type