diff options
Diffstat (limited to 'src/armnn/SerializeLayerParameters.hpp')
-rw-r--r-- | src/armnn/SerializeLayerParameters.hpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/armnn/SerializeLayerParameters.hpp b/src/armnn/SerializeLayerParameters.hpp index 9c5b82651c..f8fe5e2992 100644 --- a/src/armnn/SerializeLayerParameters.hpp +++ b/src/armnn/SerializeLayerParameters.hpp @@ -40,6 +40,11 @@ template <> struct StringifyLayerParameters<BatchToSpaceNdDescriptor> static void Serialize(ParameterStringifyFunction& fn, const BatchToSpaceNdDescriptor& desc); }; +template <> struct StringifyLayerParameters<ChannelShuffleDescriptor> +{ + static void Serialize(ParameterStringifyFunction& fn, const ChannelShuffleDescriptor& desc); +}; + template <> struct StringifyLayerParameters<ComparisonDescriptor> { static void Serialize(ParameterStringifyFunction& fn, const ComparisonDescriptor& desc); |