diff --git a/keras/src/backend/openvino/excluded_concrete_tests.txt b/keras/src/backend/openvino/excluded_concrete_tests.txt index c2ecf0371599..a7fec75c875e 100644 --- a/keras/src/backend/openvino/excluded_concrete_tests.txt +++ b/keras/src/backend/openvino/excluded_concrete_tests.txt @@ -98,7 +98,6 @@ NumpyOneInputOpsCorrectnessTest::test_meshgrid NumpyOneInputOpsCorrectnessTest::test_min NumpyOneInputOpsCorrectnessTest::test_moveaxis NumpyOneInputOpsCorrectnessTest::test_nan_to_num -NumpyOneInputOpsCorrectnessTest::test_ndim NumpyOneInputOpsCorrectnessTest::test_nonzero NumpyOneInputOpsCorrectnessTest::test_pad_float16_constant_2 NumpyOneInputOpsCorrectnessTest::test_pad_float32_constant_2 diff --git a/keras/src/backend/openvino/numpy.py b/keras/src/backend/openvino/numpy.py index f1a998b8e939..f5c88fbff63b 100644 --- a/keras/src/backend/openvino/numpy.py +++ b/keras/src/backend/openvino/numpy.py @@ -1036,7 +1036,10 @@ def nan_to_num(x, nan=0.0, posinf=None, neginf=None): def ndim(x): - raise NotImplementedError("`ndim` is not supported with openvino backend") + x = get_ov_output(x) + x_shape = ov_opset.shape_of(x).output(0) + x_dim = ov_opset.shape_of(x_shape, "i64") + return x_dim def nonzero(x):