File tree 2 files changed +3
-5
lines changed
keras/src/backend/openvino
2 files changed +3
-5
lines changed Original file line number Diff line number Diff line change @@ -39,7 +39,6 @@ NumpyDtypeTest::test_min
39
39
NumpyDtypeTest::test_moveaxis
40
40
NumpyDtypeTest::test_multiply
41
41
NumpyDtypeTest::test_nan
42
- NumpyDtypeTest::test_nonzero
43
42
NumpyDtypeTest::test_outer_
44
43
NumpyDtypeTest::test_power
45
44
NumpyDtypeTest::test_prod
@@ -98,7 +97,6 @@ NumpyOneInputOpsCorrectnessTest::test_meshgrid
98
97
NumpyOneInputOpsCorrectnessTest::test_min
99
98
NumpyOneInputOpsCorrectnessTest::test_moveaxis
100
99
NumpyOneInputOpsCorrectnessTest::test_nan_to_num
101
- NumpyOneInputOpsCorrectnessTest::test_nonzero
102
100
NumpyOneInputOpsCorrectnessTest::test_pad_float16_constant_2
103
101
NumpyOneInputOpsCorrectnessTest::test_pad_float32_constant_2
104
102
NumpyOneInputOpsCorrectnessTest::test_pad_float64_constant_2
Original file line number Diff line number Diff line change @@ -1043,9 +1043,9 @@ def ndim(x):
1043
1043
1044
1044
1045
1045
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 )
1049
1049
1050
1050
1051
1051
def not_equal (x1 , x2 ):
You can’t perform that action at this time.
0 commit comments