Skip to content

Commit 65f7e6d

Browse files
authored
[OpenVINO BACKEND] - feat: implement numpy.nonzero for openvino backend (#21163)
* feat: implement numpy.nonzero for openvino backend Signed-off-by: 11happy <[email protected]> * format code Signed-off-by: 11happy <[email protected]> --------- Signed-off-by: 11happy <[email protected]>
1 parent c7b6b42 commit 65f7e6d

File tree

2 files changed

+3
-5
lines changed

2 files changed

+3
-5
lines changed

keras/src/backend/openvino/excluded_concrete_tests.txt

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,6 @@ NumpyDtypeTest::test_min
3939
NumpyDtypeTest::test_moveaxis
4040
NumpyDtypeTest::test_multiply
4141
NumpyDtypeTest::test_nan
42-
NumpyDtypeTest::test_nonzero
4342
NumpyDtypeTest::test_outer_
4443
NumpyDtypeTest::test_power
4544
NumpyDtypeTest::test_prod
@@ -98,7 +97,6 @@ NumpyOneInputOpsCorrectnessTest::test_meshgrid
9897
NumpyOneInputOpsCorrectnessTest::test_min
9998
NumpyOneInputOpsCorrectnessTest::test_moveaxis
10099
NumpyOneInputOpsCorrectnessTest::test_nan_to_num
101-
NumpyOneInputOpsCorrectnessTest::test_nonzero
102100
NumpyOneInputOpsCorrectnessTest::test_pad_float16_constant_2
103101
NumpyOneInputOpsCorrectnessTest::test_pad_float32_constant_2
104102
NumpyOneInputOpsCorrectnessTest::test_pad_float64_constant_2

keras/src/backend/openvino/numpy.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1043,9 +1043,9 @@ def ndim(x):
10431043

10441044

10451045
def nonzero(x):
1046-
raise NotImplementedError(
1047-
"`nonzero` is not supported with openvino backend"
1048-
)
1046+
x = get_ov_output(x)
1047+
res = ov_opset.non_zero(data=x, output_type="i32").output(0)
1048+
return OpenVINOKerasTensor(res)
10491049

10501050

10511051
def not_equal(x1, x2):

0 commit comments

Comments
 (0)