Skip to content

Commit 184bfd7

Browse files
committed
[Range] also work with np.float16
1 parent 11850c1 commit 184bfd7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/qonnx/util/range_analysis.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -207,7 +207,7 @@ def calc_convtranspose_range(node, model, range_dict):
207207
def get_minmax_prototype_tensors(irange, ishp, inp_vi, i_channel_axis=1):
208208
proto_min = valueinfo_to_tensor(inp_vi)
209209
proto_max = valueinfo_to_tensor(inp_vi)
210-
if type(irange[0]) in [float, int, np.float32, np.float64, np.uint8, np.int8]:
210+
if type(irange[0]) in [float, int, np.float16, np.float32, np.float64, np.uint8, np.int8]:
211211
imin, imax = irange
212212
proto_min[...] = imin
213213
proto_max[...] = imax

0 commit comments

Comments
 (0)