diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2021-01-13 15:56:51 +0000 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2021-01-13 15:56:51 +0000 |
commit | b3021435ad91e494af01ca0778915877dc0780c0 (patch) | |
tree | 176bff8eafd8f6a2679d3569cbd640400b10a9ef /ModelToINetworkConverter.cpp | |
parent | f36e10b8947fe5f0984e7428c2d5d5d7fe18007e (diff) | |
download | android-nn-driver-b3021435ad91e494af01ca0778915877dc0780c0.tar.gz |
IVGCVSW-4417 'Serialise ArmNN Model on android-nn-driver'
* Implemented serialization of the network on android-nn-driver
!armnn:4850
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I3caf07bd4d1d2a3068c58f0b344303c4cf977ca6
Diffstat (limited to 'ModelToINetworkConverter.cpp')
-rw-r--r-- | ModelToINetworkConverter.cpp | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/ModelToINetworkConverter.cpp b/ModelToINetworkConverter.cpp index e8cf8a8b..c205a575 100644 --- a/ModelToINetworkConverter.cpp +++ b/ModelToINetworkConverter.cpp @@ -87,8 +87,9 @@ void ModelToINetworkConverter<HalPolicy>::Convert() const HalOperand& operand = getMainModel(m_Model).operands[inputIndex]; ALOGV("ModelToINetworkConverter::Convert(): GetTensorInfoForOperand(operand)"); const armnn::TensorInfo& tensor = GetTensorInfoForOperand(operand); - ALOGV("ModelToINetworkConverter::Convert(): m_Data.m_Network->AddInputLayer(i)"); - armnn::IConnectableLayer* layer = m_Data.m_Network->AddInputLayer(i); + const std::string layerName = "Input_" + std::to_string(i); + ALOGV("ModelToINetworkConverter::Convert(): m_Data.m_Network->AddInputLayer(i, layerName.c_str())"); + armnn::IConnectableLayer* layer = m_Data.m_Network->AddInputLayer(i, layerName.c_str()); ALOGV("ModelToINetworkConverter::Convert(): layer->GetOutputSlot(0)"); armnn::IOutputSlot& outputSlot = layer->GetOutputSlot(0); @@ -184,7 +185,8 @@ void ModelToINetworkConverter<HalPolicy>::Convert() uint32_t outputIndex = getMainModel(m_Model).outputIndexes[i]; const HalOperand& operand = getMainModel(m_Model).operands[outputIndex]; const armnn::TensorInfo& tensor = GetTensorInfoForOperand(operand); - armnn::IConnectableLayer* layer = m_Data.m_Network->AddOutputLayer(i); + const std::string layerName = "Output_" + std::to_string(i); + armnn::IConnectableLayer* layer = m_Data.m_Network->AddOutputLayer(i, layerName.c_str()); assert(m_Data.m_OutputSlotForOperand[outputIndex]); m_Data.m_OutputSlotForOperand[outputIndex]->Connect(layer->GetInputSlot(0)); |