diff options
Diffstat (limited to 'src/armnn/SerializeLayerParameters.cpp')
-rw-r--r-- | src/armnn/SerializeLayerParameters.cpp | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/src/armnn/SerializeLayerParameters.cpp b/src/armnn/SerializeLayerParameters.cpp index d65a7d55fa..cc59e1fad3 100644 --- a/src/armnn/SerializeLayerParameters.cpp +++ b/src/armnn/SerializeLayerParameters.cpp @@ -325,6 +325,15 @@ void StringifyLayerParameters<PreCompiledDescriptor>::Serialize(ParameterStringi fn("NumOutputSlots", std::to_string(desc.m_NumOutputSlots)); } +void StringifyLayerParameters<FusedDescriptor>::Serialize(ParameterStringifyFunction& fn, + const FusedDescriptor& desc) +{ + fn("NumInputSlots", std::to_string(desc.m_NumInputSlots)); + fn("NumOutputSlots", std::to_string(desc.m_NumOutputSlots)); + fn("PaddingMode", GetFusedTypeAsCString(desc.m_FusedKernelType)); + +} + void StringifyLayerParameters<Pooling2dDescriptor>::Serialize(ParameterStringifyFunction& fn, const Pooling2dDescriptor& desc) { |