Publishing 2019 R3 content

[platform/upstream/dldt.git] / inference-engine / src / vpu / graph_transformer / src / stages / power.cpp

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 (file)
--- 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;