Skip to content

Commit 31a81c5

Browse files
committed
Fix missing range dtype
1 parent 63bcd82 commit 31a81c5

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

keras/src/backend/openvino/numpy.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1092,6 +1092,7 @@ def median(x, axis=None, keepdims=False):
10921092
ov_opset.constant([0], Type.i32).output(0),
10931093
x_rank,
10941094
ov_opset.constant([1], Type.i32).output(0),
1095+
"i32",
10951096
).output(0)
10961097
axis_compare = ov_opset.not_equal(
10971098
ov_opset.unsqueeze(axis_as_range, 1).output(0),
@@ -1145,6 +1146,7 @@ def median(x, axis=None, keepdims=False):
11451146
ov_opset.constant([0], Type.i32).output(0),
11461147
x_rank,
11471148
ov_opset.constant([1], Type.i32).output(0),
1149+
"i32",
11481150
).output(0)
11491151
ov_axis_positive = ov_opset.gather(
11501152
axis_as_range, ov_axis, ov_opset.constant([0], Type.i32)

0 commit comments

Comments
 (0)