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

diff --git a/inference-engine/src/vpu/graph_transformer/src/stages/power.cpp b/inference-engine/src/vpu/graph_transformer/src/stages/power.cpp
index 2f83e63..9504c73 100644
--- a/inference-engine/src/vpu/graph_transformer/src/stages/power.cpp
+++ b/inference-engine/src/vpu/graph_transformer/src/stages/power.cpp
@@ -48,21 +48,19 @@ private:
 
     void propagateScaleFactorsImpl(
             const SmallVector<float>& inputScales,
-            ScalePropagationStep step) override {
-        IE_ASSERT(_inputEdges.size() == 1);
-        IE_ASSERT(_outputEdges.size() == 1);
-
+            ScalePropagationStep step,
+            StageDataInfo<float>& scaleInfo) override {
         auto power = attrs().get<float>("power");
         auto& scale = attrs().get<float>("scale");
         auto& bias = attrs().get<float>("bias");
 
         if (power != 1.0f) {
-            _scaleInfo.setInput(_inputEdges[0], 1.0f);
-            _scaleInfo.setOutput(_outputEdges[0], 1.0f);
+            scaleInfo.setInput(inputEdge(0), 1.0f);
+            scaleInfo.setOutput(outputEdge(0), 1.0f);
         } else {
             auto inputScale = inputScales[0];
 
-            _scaleInfo.setOutput(_outputEdges[0], inputScale);
+            scaleInfo.setOutput(outputEdge(0), inputScale);
 
             if (step == ScalePropagationStep::ScaleInput) {
                 scale *= inputScale;