Skip to content

Commit fbb727a

Browse files
committed
Fix missing gather axis arguements
1 parent 77f7859 commit fbb727a

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
@@ -1086,6 +1086,7 @@ def median(x, axis=None, keepdims=False):
10861086
x_rank = ov_opset.gather(
10871087
ov_opset.shape_of(x_shape_original, Type.i32).output(0),
10881088
ov_opset.constant([0], Type.i32).output(0),
1089+
ov_opset.constant([0], Type.i32).output(0),
10891090
).output(0)
10901091
axis_as_range = ov_opset.range(
10911092
ov_opset.constant([0], Type.i32).output(0),
@@ -1138,6 +1139,7 @@ def median(x, axis=None, keepdims=False):
11381139
x_rank = ov_opset.gather(
11391140
ov_opset.shape_of(x, Type.i32).output(0),
11401141
ov_opset.constant([0], Type.i32).output(0),
1142+
ov_opset.constant([0], Type.i32).output(0),
11411143
).output(0)
11421144
axis_as_range = ov_opset.range(
11431145
ov_opset.constant([0], Type.i32).output(0),

0 commit comments

Comments
 (0)