From 400c593e03af3cc9131058b3cb30b787586f69a3 Mon Sep 17 00:00:00 2001 From: Teresa Charlin Date: Fri, 4 Jun 2021 10:29:47 +0100 Subject: COMPMID-4548 Print Elementwise Descriptor on dot graph *Order the functions alphabetically Signed-off-by: Teresa Charlin Change-Id: Ia19f2727733026aefab17ebce0cabb66863a7af7 --- src/armnn/SerializeLayerParameters.hpp | 98 ++++++++++++++++++---------------- 1 file changed, 51 insertions(+), 47 deletions(-) (limited to 'src/armnn/SerializeLayerParameters.hpp') diff --git a/src/armnn/SerializeLayerParameters.hpp b/src/armnn/SerializeLayerParameters.hpp index f760df0275..9c5b82651c 100644 --- a/src/armnn/SerializeLayerParameters.hpp +++ b/src/armnn/SerializeLayerParameters.hpp @@ -25,24 +25,24 @@ struct StringifyLayerParameters static void Serialize(ParameterStringifyFunction&, const LayerParameter&) {} }; -template <> struct StringifyLayerParameters +template <> struct StringifyLayerParameters { - static void Serialize(ParameterStringifyFunction& fn, const ComparisonDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const ActivationDescriptor& desc); }; -template <> struct StringifyLayerParameters +template <> struct StringifyLayerParameters { - static void Serialize(ParameterStringifyFunction& fn, const PermuteDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const BatchNormalizationDescriptor& desc); }; -template <> struct StringifyLayerParameters +template <> struct StringifyLayerParameters { - static void Serialize(ParameterStringifyFunction& fn, const ReshapeDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const BatchToSpaceNdDescriptor& desc); }; -template <> struct StringifyLayerParameters +template <> struct StringifyLayerParameters { - static void Serialize(ParameterStringifyFunction& fn, const ActivationDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const ComparisonDescriptor& desc); }; template <> struct StringifyLayerParameters @@ -50,9 +50,9 @@ template <> struct StringifyLayerParameters static void Serialize(ParameterStringifyFunction& fn, const Convolution2dDescriptor& desc); }; -template <> struct StringifyLayerParameters +template <> struct StringifyLayerParameters { - static void Serialize(ParameterStringifyFunction& fn, const BatchNormalizationDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const DetectionPostProcessDescriptor& desc); }; template <> struct StringifyLayerParameters @@ -60,19 +60,14 @@ template <> struct StringifyLayerParameters static void Serialize(ParameterStringifyFunction& fn, const DepthwiseConvolution2dDescriptor& desc); }; -template <> struct StringifyLayerParameters -{ - static void Serialize(ParameterStringifyFunction& fn, const Pooling2dDescriptor& desc); -}; - -template <> struct StringifyLayerParameters +template <> struct StringifyLayerParameters { - static void Serialize(ParameterStringifyFunction& fn, const ReduceDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const ElementwiseUnaryDescriptor& desc); }; -template <> struct StringifyLayerParameters +template <> struct StringifyLayerParameters { - static void Serialize(ParameterStringifyFunction& fn, const SoftmaxDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const FakeQuantizationDescriptor& desc); }; template <> struct StringifyLayerParameters @@ -80,19 +75,19 @@ template <> struct StringifyLayerParameters static void Serialize(ParameterStringifyFunction& fn, const FullyConnectedDescriptor& desc); }; -template <> struct StringifyLayerParameters +template <> struct StringifyLayerParameters { - static void Serialize(ParameterStringifyFunction& fn, const OriginsDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const L2NormalizationDescriptor& desc); }; -template <> struct StringifyLayerParameters +template <> struct StringifyLayerParameters { - static void Serialize(ParameterStringifyFunction& fn, const ViewsDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const LstmDescriptor& desc); }; -template <> struct StringifyLayerParameters +template <> struct StringifyLayerParameters { - static void Serialize(ParameterStringifyFunction& fn, const DetectionPostProcessDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const MeanDescriptor& desc); }; template <> struct StringifyLayerParameters @@ -100,54 +95,58 @@ template <> struct StringifyLayerParameters static void Serialize(ParameterStringifyFunction& fn, const NormalizationDescriptor& desc); }; -template <> struct StringifyLayerParameters +template <> struct StringifyLayerParameters { - static void Serialize(ParameterStringifyFunction& fn, const L2NormalizationDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const OriginsDescriptor& desc); }; -template <> struct StringifyLayerParameters +template <> struct StringifyLayerParameters { - static void Serialize(ParameterStringifyFunction& fn, const BatchToSpaceNdDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const PadDescriptor& desc); +}; +template <> struct StringifyLayerParameters +{ + static void Serialize(ParameterStringifyFunction& fn, const PermuteDescriptor& desc); }; -template <> struct StringifyLayerParameters +template <> struct StringifyLayerParameters { - static void Serialize(ParameterStringifyFunction& fn, const FakeQuantizationDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const Pooling2dDescriptor& desc); }; -template <> struct StringifyLayerParameters +template <> struct StringifyLayerParameters { - static void Serialize(ParameterStringifyFunction& fn, const ResizeBilinearDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const PreCompiledDescriptor& desc); }; -template <> struct StringifyLayerParameters +template <> struct StringifyLayerParameters { - static void Serialize(ParameterStringifyFunction& fn, const ResizeDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const ReduceDescriptor& desc); }; -template <> struct StringifyLayerParameters +template <> struct StringifyLayerParameters { - static void Serialize(ParameterStringifyFunction& fn, const SpaceToBatchNdDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const ReshapeDescriptor& desc); }; -template <> struct StringifyLayerParameters +template <> struct StringifyLayerParameters { - static void Serialize(ParameterStringifyFunction& fn, const SpaceToDepthDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const ResizeBilinearDescriptor& desc); }; -template <> struct StringifyLayerParameters +template <> struct StringifyLayerParameters { - static void Serialize(ParameterStringifyFunction& fn, const LstmDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const ResizeDescriptor& desc); }; -template <> struct StringifyLayerParameters +template <> struct StringifyLayerParameters { - static void Serialize(ParameterStringifyFunction& fn, const MeanDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const SpaceToBatchNdDescriptor& desc); }; -template <> struct StringifyLayerParameters +template <> struct StringifyLayerParameters { - static void Serialize(ParameterStringifyFunction& fn, const PadDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const SpaceToDepthDescriptor& desc); }; template <> struct StringifyLayerParameters @@ -160,9 +159,9 @@ template <> struct StringifyLayerParameters static void Serialize(ParameterStringifyFunction& fn, const StridedSliceDescriptor& desc); }; -template <> struct StringifyLayerParameters +template <> struct StringifyLayerParameters { - static void Serialize(ParameterStringifyFunction& fn, const PreCompiledDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const SoftmaxDescriptor& desc); }; template <> struct StringifyLayerParameters @@ -175,4 +174,9 @@ template <> struct StringifyLayerParameters static void Serialize(ParameterStringifyFunction& fn, const TransposeDescriptor& desc); }; +template <> struct StringifyLayerParameters +{ + static void Serialize(ParameterStringifyFunction& fn, const ViewsDescriptor& desc); +}; + } // namespace armnn \ No newline at end of file -- cgit v1.2.1