Skip to content

Commit 22ba3f2

Browse files
author
Saurabh Singh
committed
fix
1 parent 3789a5f commit 22ba3f2

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

keras/src/backend/openvino/numpy.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1051,7 +1051,8 @@ def median(x, axis=None, keepdims=False):
10511051
elif axis is None and x.get_partial_shape().rank.get_length() > 1:
10521052
scalar_shape = ov_opset.constant([], Type.i32).output(0)
10531053
median_value = ov_opset.reshape(median_value, scalar_shape, False).output(0)
1054-
1054+
1055+
median_value = ov_opset.convert(median_value, x.get_element_type()).output(0)
10551056
return OpenVINOKerasTensor(median_value)
10561057

10571058

0 commit comments

Comments
 (0)