File tree Expand file tree Collapse file tree 2 files changed +9
-7
lines changed
keras/src/backend/openvino Expand file tree Collapse file tree 2 files changed +9
-7
lines changed Original file line number Diff line number Diff line change @@ -26,7 +26,6 @@ NumpyDtypeTest::test_dot
26
26
NumpyDtypeTest::test_einsum
27
27
NumpyDtypeTest::test_empty
28
28
NumpyDtypeTest::test_exp2
29
- NumpyDtypeTest::test_exp
30
29
NumpyDtypeTest::test_expm1
31
30
NumpyDtypeTest::test_eye
32
31
NumpyDtypeTest::test_flip
@@ -102,8 +101,8 @@ NumpyOneInputOpsCorrectnessTest::test_diag
102
101
NumpyOneInputOpsCorrectnessTest::test_diagonal
103
102
NumpyOneInputOpsCorrectnessTest::test_diff
104
103
NumpyOneInputOpsCorrectnessTest::test_dot
105
- NumpyOneInputOpsCorrectnessTest::test_exp
106
- NumpyOneInputOpsCorrectnessTest::test_expand_dims
104
+ NumpyOneInputOpsCorrectnessTest::test_exp2
105
+ NumpyOneInputOpsCorrectnessTest::test_expm1
107
106
NumpyOneInputOpsCorrectnessTest::test_flip
108
107
NumpyOneInputOpsCorrectnessTest::test_floor_divide
109
108
NumpyOneInputOpsCorrectnessTest::test_hstack
Original file line number Diff line number Diff line change @@ -533,14 +533,17 @@ def equal(x1, x2):
533
533
534
534
def exp (x ):
535
535
x = get_ov_output (x )
536
+ x_type = x .get_element_type ()
537
+ if x_type .is_integral ():
538
+ ov_type = OPENVINO_DTYPES [config .floatx ()]
539
+ x = ov_opset .convert (x , ov_type )
536
540
return OpenVINOKerasTensor (ov_opset .exp (x ).output (0 ))
537
541
538
542
539
543
def expand_dims (x , axis ):
540
- if isinstance (x , OpenVINOKerasTensor ):
541
- x = x .output
542
- else :
543
- assert False
544
+ x = get_ov_output (x )
545
+ if isinstance (axis , tuple ):
546
+ axis = list (axis )
544
547
axis = ov_opset .constant (axis , Type .i32 ).output (0 )
545
548
return OpenVINOKerasTensor (ov_opset .unsqueeze (x , axis ).output (0 ))
546
549
You can’t perform that action at this time.
0 commit comments