We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c5841a2 commit de790caCopy full SHA for de790ca
hls4ml/model/layers.py
@@ -402,9 +402,9 @@ def initialize(self):
402
else:
403
raise RuntimeError("Reshape for ONNX requires the target shape to be a second input.")
404
405
- # nones should not exist here
+ # remove Nones -- Seems to be used by pytorch parser
406
if target_shape[0] is None:
407
- raise RuntimeError(f"Unexpectedly have a None in {target_shape=}")
+ target_shape = target_shape[1:]
408
409
# take care of -1 shapes
410
shape = self._infer_output_shape(input_shape, target_shape)
0 commit comments