[OpenVINO backend] support repeat

Author: Mohamed-Ashraf273

keras/src/backend/openvino/excluded_concrete_tests.txt

@@ -45,7 +45,6 @@ NumpyDtypeTest::test_multiply
 NumpyDtypeTest::test_power
 NumpyDtypeTest::test_prod
 NumpyDtypeTest::test_quantile
-NumpyDtypeTest::test_repeat
 NumpyDtypeTest::test_roll
 NumpyDtypeTest::test_round
 NumpyDtypeTest::test_searchsorted
@@ -106,7 +105,6 @@ NumpyOneInputOpsCorrectnessTest::test_pad_uint8_constant_2
 NumpyOneInputOpsCorrectnessTest::test_pad_int32_constant_2
 NumpyOneInputOpsCorrectnessTest::test_prod
 NumpyOneInputOpsCorrectnessTest::test_real
-NumpyOneInputOpsCorrectnessTest::test_repeat
 NumpyOneInputOpsCorrectnessTest::test_reshape
 NumpyOneInputOpsCorrectnessTest::test_roll
 NumpyOneInputOpsCorrectnessTest::test_round

keras/src/backend/openvino/numpy.py

@@ -1291,8 +1291,48 @@ def reciprocal(x):
 
 
 def repeat(x, repeats, axis=None):
-    raise NotImplementedError("`repeat` is not supported with openvino backend")
+    x = get_ov_output(x)
+
+    if axis is not None and axis < 0:
+        axis += len(x.get_partial_shape())
 
+    if axis is None:
+        x = ov_opset.reshape(x, ov_opset.constant([-1], Type.i32), special_zero=False).output(0)
+        axis = 0
+
+    if isinstance(repeats, (int, np.integer)) or (
+        isinstance(repeats, np.ndarray) and repeats.ndim == 1 and repeats.size == 1
+    ):
+        repeats_val = int(repeats) if isinstance(repeats, np.ndarray) else repeats
+        input_shape = ov_opset.shape_of(x, Type.i32).output(0)
+        dim_len = ov_opset.gather(input_shape, ov_opset.constant([axis], Type.i32), ov_opset.constant(0, Type.i32)).output(0)
+        dim_len = ov_opset.squeeze(dim_len, ov_opset.constant([0], Type.i32)).output(0)
+        idx_range = ov_opset.range(
+            ov_opset.constant(0, Type.i32),
+            dim_len,
+            ov_opset.constant(1, Type.i32),
+            output_type=Type.i32,
+        ).output(0)
+        idx_range = ov_opset.unsqueeze(idx_range, ov_opset.constant([1], Type.i32)).output(0)
+        tiled = ov_opset.tile(idx_range, ov_opset.constant([1, repeats_val], Type.i32)).output(0)
+        idx = ov_opset.reshape(tiled, ov_opset.constant([-1], Type.i32), special_zero=False).output(0)
+        result = ov_opset.gather(x, idx, ov_opset.constant(axis, Type.i32)).output(0)
+        return OpenVINOKerasTensor(result)
+
+    repeats_np = np.array(repeats)
+    if repeats_np.ndim != 1:
+        raise NotImplementedError("Only 1D repeats arrays are supported.")
+    input_shape = ov_opset.shape_of(x, Type.i32).output(0)
+    axis_len = ov_opset.gather(input_shape, ov_opset.constant([axis], Type.i32), ov_opset.constant(0, Type.i32)).output(0)
+    axis_len_val = int(axis_len.get_vector()[0]) if hasattr(axis_len, "get_vector") else x.get_partial_shape()[axis]
+    if axis_len_val != len(repeats_np):
+        raise ValueError("repeats length does not match axis length")
+
+    gather_indices = np.concatenate([np.full(r, i, dtype=np.int32) for i, r in enumerate(repeats_np) if r > 0])
+    gather_indices_ov = ov_opset.constant(gather_indices, Type.i32).output(0)
+    result = ov_opset.gather(x, gather_indices_ov, ov_opset.constant(axis, Type.i32)).output(0)
+    return OpenVINOKerasTensor(result)
+        
 
 def reshape(x, newshape):
     x = get_ov_output(x)