Fix range start is scalar

Author: alungriffith

keras/src/backend/openvino/numpy.py

@@ -1088,10 +1088,13 @@ def median(x, axis=None, keepdims=False):
             ov_opset.constant([0], Type.i32).output(0),
             ov_opset.constant([0], Type.i32).output(0),
         ).output(0)
+        x_rank_scalar = ov_opset.squeeze(
+            x_rank, ov_opset.constant([0], Type.i32).output(0)
+        ).output(0)
         axis_as_range = ov_opset.range(
-            ov_opset.constant([0], Type.i32).output(0),
-            x_rank,
-            ov_opset.constant([1], Type.i32).output(0),
+            ov_opset.constant(0, Type.i32).output(0),
+            x_rank_scalar,
+            ov_opset.constant(1, Type.i32).output(0),
             "i32",
         ).output(0)
         axis_compare = ov_opset.not_equal(
@@ -1142,10 +1145,13 @@ def median(x, axis=None, keepdims=False):
             ov_opset.constant([0], Type.i32).output(0),
             ov_opset.constant([0], Type.i32).output(0),
         ).output(0)
+        x_rank_scalar = ov_opset.squeeze(
+            x_rank, ov_opset.constant([0], Type.i32).output(0)
+        ).output(0)
         axis_as_range = ov_opset.range(
-            ov_opset.constant([0], Type.i32).output(0),
-            x_rank,
-            ov_opset.constant([1], Type.i32).output(0),
+            ov_opset.constant(0, Type.i32).output(0),
+            x_rank_scalar,
+            ov_opset.constant(1, Type.i32).output(0),
             "i32",
         ).output(0)
         ov_axis_positive = ov_opset.gather(