X-Git-Url: http://review.tizen.org/git/?a=blobdiff_plain;f=inference-engine%2Fsrc%2Fvpu%2Fgraph_transformer%2Fsrc%2Fstages%2Fscale.cpp;h=af07f88300d686682c8344afc997efda23a572bf;hb=0923303e0201c5b59386ab146d0e30b2ef79272d;hp=eec26944954d55ae82c951f1dc970bc35e5eddf7;hpb=ba6e22b1b5ee4cbefcc30e8d9493cddb0bb3dfdf;p=platform%2Fupstream%2Fdldt.git

diff --git a/inference-engine/src/vpu/graph_transformer/src/stages/scale.cpp b/inference-engine/src/vpu/graph_transformer/src/stages/scale.cpp
index eec2694..af07f88 100644
--- a/inference-engine/src/vpu/graph_transformer/src/stages/scale.cpp
+++ b/inference-engine/src/vpu/graph_transformer/src/stages/scale.cpp
@@ -23,17 +23,15 @@ private:
 
     void propagateScaleFactorsImpl(
             const SmallVector<float>& inputScales,
-            ScalePropagationStep step) override {
-        IE_ASSERT(_inputEdges.size() == 2 || _inputEdges.size() == 3);
-        IE_ASSERT(_outputEdges.size() == 1);
-
+            ScalePropagationStep step,
+            StageDataInfo<float>& scaleInfo) override {
         auto inputScale = inputScales[0];
 
-        _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);
         }
-        _scaleInfo.setOutput(_outputEdges[0], inputScale);
+        scaleInfo.setOutput(outputEdge(0), inputScale);
     }
 
     void serializeParamsImpl(BlobSerializer&) const override {