[LPT] FuseSubtractToFakeQuantize after VariadicSplit

eshoguli · Sep 13, 2020 · 8c44d5b · 8c44d5b
1 parent 45186c5
commit 8c44d5b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/.../src/transformations/src/transformations/low_precision/fuse_subtract_to_fake_quantize.cpp b/.../src/transformations/src/transformations/low_precision/fuse_subtract_to_fake_quantize.cpp
@@ -39,8 +39,12 @@ bool FuseSubtractToFakeQuantizeTransformation::transform(TransformationContext&
     outputLowConst = fold<opset1::Subtract>(outputLowConst, value);
     outputHighConst = fold<opset1::Subtract>(outputHighConst, value);
 
+    const auto fakeQuantizeParent = fakeQuantize->get_input_node_shared_ptr(0);
+    const size_t parentIndex = NetworkHelper::getInputIndex(fakeQuantizeParent, fakeQuantize);
+
     auto newFakeQuantize = std::make_shared<op::TypeRelaxed<opset1::FakeQuantize>>(
-        opset1::FakeQuantize(fakeQuantize->get_input_node_shared_ptr(0),
+        opset1::FakeQuantize(
+            fakeQuantizeParent->output(parentIndex),
             fakeQuantize->input_value(1),
             fakeQuantize->input_value(2),
             outputLowConst,