diff options
Diffstat (limited to 'src/armnnDeserializer')
-rw-r--r-- | src/armnnDeserializer/Deserializer.cpp | 4 | ||||
-rw-r--r-- | src/armnnDeserializer/test/DeserializeActivation.cpp | 2 |
2 files changed, 3 insertions, 3 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()); diff --git a/src/armnnDeserializer/test/DeserializeActivation.cpp b/src/armnnDeserializer/test/DeserializeActivation.cpp index ad03dd662b..9e98893038 100644 --- a/src/armnnDeserializer/test/DeserializeActivation.cpp +++ b/src/armnnDeserializer/test/DeserializeActivation.cpp @@ -71,7 +71,7 @@ struct ActivationFixture : public ParserFlatbuffersSerializeFixture descriptor: { a: )" + a + R"(, b: )" + b + R"(, - function: )" + activationType + R"( + activationFunction: )" + activationType + R"( }, }, }, |