diff --git a/src/finn/transformation/streamline/reorder.py b/src/finn/transformation/streamline/reorder.py index e36de2aa54f7b260016dd0b1b93bccbfca9d72b9..2fd8f36d6365287706947a6c78653baa1c124ff9 100644 --- a/src/finn/transformation/streamline/reorder.py +++ b/src/finn/transformation/streamline/reorder.py @@ -735,6 +735,12 @@ class MakeScaleResizeNHWC(Transformation): for n in graph.node: node_ind += 1 if n.op_type == "Upsample" or n.op_type == "Resize": + if model.get_tensor_layout(n.input[0]) != DataLayout.NCHW: + warnings.warn( + "%s: Input not NCHW. Can't operate transformation on node." + % n.name + ) + continue consumer = model.find_consumer(n.output[0]) producer = model.find_producer(n.input[0]) if n.op_type == "Upsample":