diff options
author | Saoirse Stewart <saoirse.stewart@arm.com> | 2019-02-14 14:57:37 +0000 |
---|---|---|
committer | Saoirse Stewart Arm <saoirse.stewart@arm.com> | 2019-02-15 08:57:12 +0000 |
commit | 3fcef20ff9b6fec5b853bc8166db8b9ae88f9f39 (patch) | |
tree | de5ca9dc47933614e6d95438a5169cbbbad9ca85 /src/armnnDeserializeParser/DeserializeParser.cpp | |
parent | 5b2159e43a5e078053e706883e3af44ae733b5b9 (diff) | |
download | armnn-3fcef20ff9b6fec5b853bc8166db8b9ae88f9f39.tar.gz |
IVGCVSW-2642 Fix layerBindingId issue
Change-Id: I24fef25ab12df07b98d78de98563ee4a2b9977ba
Signed-off-by: Saoirse Stewart <saoirse.stewart@arm.com>
Diffstat (limited to 'src/armnnDeserializeParser/DeserializeParser.cpp')
-rw-r--r-- | src/armnnDeserializeParser/DeserializeParser.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
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); } |