Skip to content

Commit 13906da

Browse files
committed
Fix: handling scalar vs. array scaling factor
1 parent 29e6843 commit 13906da

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/qonnx/transformation/resize_conv_to_deconv.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,8 @@ def apply(self, model):
111111
[q_w_name, q_s_name, q_zp_name, q_bw_name] = weight_prod.input
112112
W_conv = model.get_initializer(q_w_name)
113113
W_scale = model.get_initializer(q_s_name)
114-
W_scale = np.moveaxis(W_scale, 0, 1)
114+
if isinstance(W_scale, np.ndarray) and W_scale.ndim > 0:
115+
W_scale = np.moveaxis(W_scale, 0, 1)
115116
W_zeropt = model.get_initializer(q_zp_name)
116117
W_bitwidth = model.get_initializer(q_bw_name)
117118
W_signed = get_by_name(weight_prod.attribute, "signed").i

0 commit comments

Comments
 (0)