Skip to content

Commit 67cb539

Browse files
feat: logaddexp for openvino backend
1 parent 90c8da6 commit 67cb539

File tree

1 file changed

+12
-3
lines changed

1 file changed

+12
-3
lines changed

keras/src/backend/openvino/numpy.py

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1021,9 +1021,18 @@ def log2(x):
10211021

10221022

10231023
def logaddexp(x1, x2):
1024-
raise NotImplementedError(
1025-
"`logaddexp` is not supported with openvino backend"
1026-
)
1024+
x1 = get_ov_output(x1)
1025+
x2 = get_ov_output(x2)
1026+
1027+
x1, x2 = _align_operand_types(x1, x2, "logaddexp()")
1028+
1029+
max_val = ov_opset.maximum(x1, x2).output(0)
1030+
abs_diff = ov_opset.abs(ov_opset.subtract(x1, x2).output(0)).output(0)
1031+
neg_abs_diff = ov_opset.negative(abs_diff).output(0)
1032+
exp_neg_abs = ov_opset.exp(neg_abs_diff).output(0)
1033+
log1p_exp = ov_opset.log1p(exp_neg_abs).output(0)
1034+
result = ov_opset.add(max_val, log1p_exp).output(0)
1035+
return OpenVINOKerasTensor(result)
10271036

10281037

10291038
def logical_and(x1, x2):

0 commit comments

Comments
 (0)