@@ -1088,13 +1088,10 @@ def median(x, axis=None, keepdims=False):
1088
1088
ov_opset .constant ([0 ], Type .i32 ).output (0 ),
1089
1089
ov_opset .constant ([0 ], Type .i32 ).output (0 ),
1090
1090
).output (0 )
1091
- x_rank_scalar = ov_opset .squeeze (
1092
- x_rank , ov_opset .constant ([0 ], Type .i32 ).output (0 )
1093
- ).output (0 )
1094
1091
axis_as_range = ov_opset .range (
1095
- ov_opset .constant (0 , Type .i32 ).output (0 ),
1096
- x_rank_scalar ,
1097
- ov_opset .constant (1 , Type .i32 ).output (0 ),
1092
+ ov_opset .constant ([ 0 ] , Type .i32 ).output (0 ),
1093
+ x_rank ,
1094
+ ov_opset .constant ([ 1 ] , Type .i32 ).output (0 ),
1098
1095
"i32" ,
1099
1096
).output (0 )
1100
1097
axis_compare = ov_opset .not_equal (
@@ -1145,13 +1142,10 @@ def median(x, axis=None, keepdims=False):
1145
1142
ov_opset .constant ([0 ], Type .i32 ).output (0 ),
1146
1143
ov_opset .constant ([0 ], Type .i32 ).output (0 ),
1147
1144
).output (0 )
1148
- x_rank_scalar = ov_opset .squeeze (
1149
- x_rank , ov_opset .constant ([0 ], Type .i32 ).output (0 )
1150
- ).output (0 )
1151
1145
axis_as_range = ov_opset .range (
1152
- ov_opset .constant (0 , Type .i32 ).output (0 ),
1153
- x_rank_scalar ,
1154
- ov_opset .constant (1 , Type .i32 ).output (0 ),
1146
+ ov_opset .constant ([ 0 ] , Type .i32 ).output (0 ),
1147
+ x_rank ,
1148
+ ov_opset .constant ([ 1 ] , Type .i32 ).output (0 ),
1155
1149
"i32" ,
1156
1150
).output (0 )
1157
1151
ov_axis_positive = ov_opset .gather (
0 commit comments