From fd0cae3153dd7844b1b14adda8571e4588ed762c Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Mon, 8 Nov 2021 17:18:31 +0000 Subject: IVGCVSW-5636 'Implement NNAPI caching functions' * Serialize/Deserialize axis parameter on Softmax Descriptor. Signed-off-by: Sadik Armagan Change-Id: I1bba0dd7d78f9f211f77115dcefccf24a4a61b39 --- src/armnnDeserializer/Deserializer.cpp | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'src/armnnDeserializer') diff --git a/src/armnnDeserializer/Deserializer.cpp b/src/armnnDeserializer/Deserializer.cpp index bfd4f6b560..8b1e9b970c 100644 --- a/src/armnnDeserializer/Deserializer.cpp +++ b/src/armnnDeserializer/Deserializer.cpp @@ -1652,7 +1652,8 @@ void IDeserializer::DeserializerImpl::ParseFill(GraphPtr graph, unsigned int lay CHECK_VALID_SIZE(outputs.size(), 1); auto layerName = GetLayerName(graph, layerIndex); - armnn::FillDescriptor descriptor(1.0f); + armnn::FillDescriptor descriptor; + descriptor.m_Value = graph->layers()->Get(layerIndex)->layer_as_FillLayer()->descriptor()->value(); IConnectableLayer* layer = m_Network->AddFillLayer(descriptor, layerName.c_str()); armnn::TensorInfo outputTensorInfo = ToTensorInfo(outputs[0]); @@ -2484,6 +2485,7 @@ void IDeserializer::DeserializerImpl::ParseSoftmax(GraphPtr graph, unsigned int armnn::SoftmaxDescriptor descriptor; descriptor.m_Beta = graph->layers()->Get(layerIndex)->layer_as_SoftmaxLayer()->descriptor()->beta(); + descriptor.m_Axis = graph->layers()->Get(layerIndex)->layer_as_SoftmaxLayer()->descriptor()->axis(); auto layerName = GetLayerName(graph, layerIndex); IConnectableLayer* layer = m_Network->AddSoftmaxLayer(descriptor, layerName.c_str()); -- cgit v1.2.1