From 8c44d5b1eafd5461872a002c39b5ff63beab8460 Mon Sep 17 00:00:00 2001
From: Edward Shogulin <edward.shogulin@intel.com>
Date: Sun, 13 Sep 2020 21:10:06 +0300
Subject: [PATCH] [LPT] FuseSubtractToFakeQuantize after VariadicSplit

---
 .../low_precision/fuse_subtract_to_fake_quantize.cpp        | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/inference-engine/src/transformations/src/transformations/low_precision/fuse_subtract_to_fake_quantize.cpp b/inference-engine/src/transformations/src/transformations/low_precision/fuse_subtract_to_fake_quantize.cpp
index 1cd75c458b6787..d59535963a3b0a 100644
--- a/inference-engine/src/transformations/src/transformations/low_precision/fuse_subtract_to_fake_quantize.cpp
+++ b/inference-engine/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,