diff options
| -rw-r--r-- | src/armnnDeserializeParser/DeserializeParser.cpp | 4 | ||||
| -rw-r--r-- | src/armnnDeserializeParser/test/DeserializeAdd.cpp | 2 | ||||
| -rw-r--r-- | src/armnnDeserializeParser/test/DeserializeMultiplication.cpp | 2 |
3 files changed, 4 insertions, 4 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); } diff --git a/src/armnnDeserializeParser/test/DeserializeAdd.cpp b/src/armnnDeserializeParser/test/DeserializeAdd.cpp index 03d50e3fe0..f0b85905b3 100644 --- a/src/armnnDeserializeParser/test/DeserializeAdd.cpp +++ b/src/armnnDeserializeParser/test/DeserializeAdd.cpp @@ -99,7 +99,7 @@ struct AddFixture : public ParserFlatbuffersSerializeFixture layer_type: "OutputLayer", layer: { base:{ - layerBindingId: 3, + layerBindingId: 0, base: { index: 3, layerName: "OutputLayer", diff --git a/src/armnnDeserializeParser/test/DeserializeMultiplication.cpp b/src/armnnDeserializeParser/test/DeserializeMultiplication.cpp index f8eff1697a..f69413b223 100644 --- a/src/armnnDeserializeParser/test/DeserializeMultiplication.cpp +++ b/src/armnnDeserializeParser/test/DeserializeMultiplication.cpp @@ -99,7 +99,7 @@ struct MultiplicationFixture : public ParserFlatbuffersSerializeFixture layer_type: "OutputLayer", layer: { base:{ - layerBindingId: 3, + layerBindingId: 0, base: { index: 3, layerName: "OutputLayer", |