aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/SerializeLayerParameters.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/SerializeLayerParameters.cpp')
-rw-r--r--src/armnn/SerializeLayerParameters.cpp9
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)
{