From 3fcef20ff9b6fec5b853bc8166db8b9ae88f9f39 Mon Sep 17 00:00:00 2001
From: Saoirse Stewart <saoirse.stewart@arm.com>
Date: Thu, 14 Feb 2019 14:57:37 +0000
Subject: IVGCVSW-2642 Fix layerBindingId issue

Change-Id: I24fef25ab12df07b98d78de98563ee4a2b9977ba
Signed-off-by: Saoirse Stewart <saoirse.stewart@arm.com>
---
 src/armnnDeserializeParser/DeserializeParser.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'src/armnnDeserializeParser/DeserializeParser.cpp')

diff --git a/src/armnnDeserializeParser/DeserializeParser.cpp b/src/armnnDeserializeParser/DeserializeParser.cpp
index 0368ccabf2..123fc9366c 100644
--- a/src/armnnDeserializeParser/DeserializeParser.cpp
+++ b/src/armnnDeserializeParser/DeserializeParser.cpp
@@ -465,7 +465,7 @@ void DeserializeParser::SetupInputLayers()
     for (auto const& input : inputs)
     {
         IConnectableLayer* layer =
-            m_Network->AddInputLayer(static_cast<int>(input->index()), input->layerName()->c_str());
+            m_Network->AddInputLayer(GetBindingLayerInfo(m_Graph, input->index()), input->layerName()->c_str());
 
         auto tensorInfo = ToTensorInfo(input->outputSlots()->Get(0)->tensorInfo());
         layer->GetOutputSlot(0).SetTensorInfo(tensorInfo);
@@ -481,7 +481,7 @@ void DeserializeParser::SetupOutputLayers()
     for (auto const& output : outputs)
     {
         IConnectableLayer* layer =
-            m_Network->AddOutputLayer(static_cast<int>(output->index()), output->layerName()->c_str());
+            m_Network->AddOutputLayer(GetBindingLayerInfo(m_Graph, output->index()), output->layerName()->c_str());
 
         RegisterInputSlots(output->index(), layer);
     }
-- 
cgit v1.2.1