[OpenVINO backend] support top_k

Author: Mohamed-Ashraf273

keras/src/backend/openvino/excluded_concrete_tests.txt

@@ -186,3 +186,52 @@ CoreOpsCorrectnessTest::test_slice_update
 CoreOpsCorrectnessTest::test_switch 
 CoreOpsCorrectnessTest::test_unstack
 CoreOpsCorrectnessTest::test_vectorized_map
+ExtractSequencesOpTest::test_extract_sequences_call
+InTopKTest::test_in_top_k_call
+MathOpsCorrectnessTest::test_erfinv_operation_basic
+MathOpsCorrectnessTest::test_erfinv_operation_dtype
+MathOpsCorrectnessTest::test_erfinv_operation_edge_cases
+MathOpsCorrectnessTest::test_extract_sequences
+MathOpsCorrectnessTest::test_fft
+MathOpsCorrectnessTest::test_fft2
+MathOpsCorrectnessTest::test_ifft2
+MathOpsCorrectnessTest::test_in_top_k
+MathOpsCorrectnessTest::test_irfft0
+MathOpsCorrectnessTest::test_irfft1
+MathOpsCorrectnessTest::test_irfft2
+MathOpsCorrectnessTest::test_istft0
+MathOpsCorrectnessTest::test_istft1
+MathOpsCorrectnessTest::test_istft2
+MathOpsCorrectnessTest::test_istft3
+MathOpsCorrectnessTest::test_istft4
+MathOpsCorrectnessTest::test_istft5
+MathOpsCorrectnessTest::test_istft6
+MathOpsCorrectnessTest::test_logdet
+MathOpsCorrectnessTest::test_logsumexp
+MathOpsCorrectnessTest::test_rfft0
+MathOpsCorrectnessTest::test_rfft1
+MathOpsCorrectnessTest::test_rfft2
+MathOpsCorrectnessTest::test_segment_reduce0
+MathOpsCorrectnessTest::test_segment_reduce1
+MathOpsCorrectnessTest::test_segment_reduce2
+MathOpsCorrectnessTest::test_segment_reduce3
+MathOpsCorrectnessTest::test_segment_reduce_explicit_num_segments0
+MathOpsCorrectnessTest::test_segment_reduce_explicit_num_segments1
+MathOpsCorrectnessTest::test_segment_reduce_explicit_num_segments2
+MathOpsCorrectnessTest::test_segment_reduce_explicit_num_segments3
+MathOpsCorrectnessTest::test_segment_reduce_explicit_num_segments4
+MathOpsCorrectnessTest::test_segment_reduce_explicit_num_segments5
+MathOpsCorrectnessTest::test_stft0
+MathOpsCorrectnessTest::test_stft1
+MathOpsCorrectnessTest::test_stft2
+MathOpsCorrectnessTest::test_stft3
+MathOpsCorrectnessTest::test_stft4
+MathOpsCorrectnessTest::test_stft5
+MathOpsCorrectnessTest::test_stft6
+SegmentSumTest::test_segment_sum_call
+SegmentMaxTest::test_segment_max_call
+TestMathErrors::test_invalid_fft_length
+TestMathErrors::test_istft_invalid_window_shape_2D_inputs
+TestMathErrors::test_stft_invalid_input_type
+TestMathErrors::test_stft_invalid_window
+TestMathErrors::test_stft_invalid_window_shape

keras/src/backend/openvino/excluded_tests.txt

@@ -29,7 +29,6 @@ keras/src/metrics
 keras/src/models
 keras/src/ops/image_test.py
 keras/src/ops/linalg_test.py
-keras/src/ops/math_test.py
 keras/src/ops/nn_test.py
 keras/src/optimizers
 keras/src/quantizers

keras/src/backend/openvino/math.py

@@ -18,7 +18,14 @@ def segment_max(data, segment_ids, num_segments=None, sorted=False):
 
 
 def top_k(x, k, sorted=True):
-    raise NotImplementedError("`top_k` is not supported with openvino backend")
+    x = get_ov_output(x)
+    k_tensor = ov_opset.constant(k, dtype=Type.i32)
+    axis = -1
+    sort_type = "value" if sorted else "none"
+    topk_node = ov_opset.topk(x, k_tensor, axis, "max", sort_type)
+    values = topk_node.output(0)
+    indices = topk_node.output(1)
+    return OpenVINOKerasTensor(values), OpenVINOKerasTensor(indices)
 
 
 def in_top_k(targets, predictions, k):