Fix: Calculate input/output size also when there is no stage in/out phase.