diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2021-11-08 17:18:31 +0000 |
---|---|---|
committer | David Monahan <david.monahan@arm.com> | 2021-11-08 20:05:43 +0000 |
commit | fd0cae3153dd7844b1b14adda8571e4588ed762c (patch) | |
tree | 03995f2f6395c4a8f8152ede6eccc5ab3bad4b59 /src/armnnDeserializer | |
parent | 5b8093c17044e8eaaaa42d96ba4902dee5791be4 (diff) | |
download | armnn-fd0cae3153dd7844b1b14adda8571e4588ed762c.tar.gz |
IVGCVSW-5636 'Implement NNAPI caching functions'
* Serialize/Deserialize axis parameter on Softmax Descriptor.
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I1bba0dd7d78f9f211f77115dcefccf24a4a61b39
Diffstat (limited to 'src/armnnDeserializer')
-rw-r--r-- | src/armnnDeserializer/Deserializer.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
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()); |