- _scaleInfo.setInput(_inputEdges[1], step == ScalePropagationStep::Propagate ? 1.0f : inputScale);
- if (_inputEdges.size() == 3) {
- _scaleInfo.setInput(_inputEdges[2], inputScale);
+ scaleInfo.setInput(inputEdge(1), step == ScalePropagationStep::Propagate ? 1.0f : inputScale);
+ if (numInputs() == 3) {
+ scaleInfo.setInput(inputEdge(2), inputScale);