Skip to content

Commit ff8b026

Browse files
committed
Fixed missing output and dtype issues
1 parent 734e05f commit ff8b026

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

keras/src/backend/openvino/numpy.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1147,7 +1147,9 @@ def median(x, axis=None, keepdims=False):
11471147
else:
11481148
ov_axis_positive = ov_axis
11491149

1150-
k_scalar = ov_opset.squeeze(k_value, ov_opset.constant([0], Type.i32).output(0)).output(0)
1150+
k_scalar = ov_opset.squeeze(
1151+
k_value, ov_opset.constant([0], Type.i32).output(0)
1152+
).output(0)
11511153
x_sorted = ov_opset.topk(
11521154
x, k_scalar, axis, "min", "value", stable=True
11531155
).output(0)
@@ -1172,12 +1174,13 @@ def median(x, axis=None, keepdims=False):
11721174
).output(0)
11731175

11741176
median_odd = med_0
1177+
median_type = med_0.get_element_type()
11751178
median_even = ov_opset.divide(
11761179
ov_opset.add(med_1, med_0).output(0),
1177-
ov_opset.constant([2], Type.f32),
1178-
)
1180+
ov_opset.constant([2], median_type),
1181+
).output(0)
11791182

1180-
median_eval = ov_opset.select(is_even, median_even, median_odd)
1183+
median_eval = ov_opset.select(is_even, median_even, median_odd).output(0)
11811184

11821185
if keepdims:
11831186
if flattened:

0 commit comments

Comments
 (0)