diff options
Diffstat (limited to 'src/armnnDeserializer/Deserializer.cpp')
-rw-r--r-- | src/armnnDeserializer/Deserializer.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/armnnDeserializer/Deserializer.cpp b/src/armnnDeserializer/Deserializer.cpp index d887357ef1..c85282f418 100644 --- a/src/armnnDeserializer/Deserializer.cpp +++ b/src/armnnDeserializer/Deserializer.cpp @@ -910,7 +910,7 @@ void Deserializer::ParseActivation(GraphPtr graph, unsigned int layerIndex) auto serializerDescriptor = serializerLayer->descriptor(); armnn::ActivationDescriptor descriptor; - descriptor.m_Function = ToActivationFunction(serializerDescriptor->function()); + descriptor.m_Function = ToActivationFunction(serializerDescriptor->activationFunction()); descriptor.m_A = serializerDescriptor->a(); descriptor.m_B = serializerDescriptor->b(); @@ -957,7 +957,7 @@ void Deserializer::ParseArgMinMax(GraphPtr graph, unsigned int layerIndex) auto serializerDescriptor = serializerLayer->descriptor(); armnn::ArgMinMaxDescriptor descriptor; - descriptor.m_Function = ToArgMinMaxFunction(serializerDescriptor->function()); + descriptor.m_Function = ToArgMinMaxFunction(serializerDescriptor->argMinMaxFunction()); descriptor.m_Axis = serializerDescriptor->axis(); auto layerName = GetLayerName(graph, layerIndex); IConnectableLayer* layer = m_Network->AddArgMinMaxLayer(descriptor, layerName.c_str()); |