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.cpp7
1 files changed, 7 insertions, 0 deletions
diff --git a/src/armnn/SerializeLayerParameters.cpp b/src/armnn/SerializeLayerParameters.cpp
index 16ffb8fd5d..73e0cbce78 100644
--- a/src/armnn/SerializeLayerParameters.cpp
+++ b/src/armnn/SerializeLayerParameters.cpp
@@ -62,6 +62,13 @@ void StringifyLayerParameters<BatchToSpaceNdDescriptor>::Serialize(ParameterStri
fn("DataLayout", GetDataLayoutName(desc.m_DataLayout));
}
+void StringifyLayerParameters<ChannelShuffleDescriptor>::Serialize(ParameterStringifyFunction& fn,
+ const ChannelShuffleDescriptor& desc)
+{
+ fn("Axis", std::to_string(desc.m_Axis));
+ fn("NumGroups", std::to_string(desc.m_NumGroups));
+}
+
void StringifyLayerParameters<ComparisonDescriptor>::Serialize(ParameterStringifyFunction& fn,
const ComparisonDescriptor& desc)
{