diff options
Diffstat (limited to 'src/armnn/SerializeLayerParameters.hpp')
-rw-r--r-- | src/armnn/SerializeLayerParameters.hpp | 98 |
1 files changed, 51 insertions, 47 deletions
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<ComparisonDescriptor> +template <> struct StringifyLayerParameters<ActivationDescriptor> { - static void Serialize(ParameterStringifyFunction& fn, const ComparisonDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const ActivationDescriptor& desc); }; -template <> struct StringifyLayerParameters<PermuteDescriptor> +template <> struct StringifyLayerParameters<BatchNormalizationDescriptor> { - static void Serialize(ParameterStringifyFunction& fn, const PermuteDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const BatchNormalizationDescriptor& desc); }; -template <> struct StringifyLayerParameters<ReshapeDescriptor> +template <> struct StringifyLayerParameters<BatchToSpaceNdDescriptor> { - static void Serialize(ParameterStringifyFunction& fn, const ReshapeDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const BatchToSpaceNdDescriptor& desc); }; -template <> struct StringifyLayerParameters<ActivationDescriptor> +template <> struct StringifyLayerParameters<ComparisonDescriptor> { - static void Serialize(ParameterStringifyFunction& fn, const ActivationDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const ComparisonDescriptor& desc); }; template <> struct StringifyLayerParameters<Convolution2dDescriptor> @@ -50,9 +50,9 @@ template <> struct StringifyLayerParameters<Convolution2dDescriptor> static void Serialize(ParameterStringifyFunction& fn, const Convolution2dDescriptor& desc); }; -template <> struct StringifyLayerParameters<BatchNormalizationDescriptor> +template <> struct StringifyLayerParameters<DetectionPostProcessDescriptor> { - static void Serialize(ParameterStringifyFunction& fn, const BatchNormalizationDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const DetectionPostProcessDescriptor& desc); }; template <> struct StringifyLayerParameters<DepthwiseConvolution2dDescriptor> @@ -60,19 +60,14 @@ template <> struct StringifyLayerParameters<DepthwiseConvolution2dDescriptor> static void Serialize(ParameterStringifyFunction& fn, const DepthwiseConvolution2dDescriptor& desc); }; -template <> struct StringifyLayerParameters<Pooling2dDescriptor> -{ - static void Serialize(ParameterStringifyFunction& fn, const Pooling2dDescriptor& desc); -}; - -template <> struct StringifyLayerParameters<ReduceDescriptor> +template <> struct StringifyLayerParameters<ElementwiseUnaryDescriptor> { - static void Serialize(ParameterStringifyFunction& fn, const ReduceDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const ElementwiseUnaryDescriptor& desc); }; -template <> struct StringifyLayerParameters<SoftmaxDescriptor> +template <> struct StringifyLayerParameters<FakeQuantizationDescriptor> { - static void Serialize(ParameterStringifyFunction& fn, const SoftmaxDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const FakeQuantizationDescriptor& desc); }; template <> struct StringifyLayerParameters<FullyConnectedDescriptor> @@ -80,19 +75,19 @@ template <> struct StringifyLayerParameters<FullyConnectedDescriptor> static void Serialize(ParameterStringifyFunction& fn, const FullyConnectedDescriptor& desc); }; -template <> struct StringifyLayerParameters<OriginsDescriptor> +template <> struct StringifyLayerParameters<L2NormalizationDescriptor> { - static void Serialize(ParameterStringifyFunction& fn, const OriginsDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const L2NormalizationDescriptor& desc); }; -template <> struct StringifyLayerParameters<ViewsDescriptor> +template <> struct StringifyLayerParameters<LstmDescriptor> { - static void Serialize(ParameterStringifyFunction& fn, const ViewsDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const LstmDescriptor& desc); }; -template <> struct StringifyLayerParameters<DetectionPostProcessDescriptor> +template <> struct StringifyLayerParameters<MeanDescriptor> { - static void Serialize(ParameterStringifyFunction& fn, const DetectionPostProcessDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const MeanDescriptor& desc); }; template <> struct StringifyLayerParameters<NormalizationDescriptor> @@ -100,54 +95,58 @@ template <> struct StringifyLayerParameters<NormalizationDescriptor> static void Serialize(ParameterStringifyFunction& fn, const NormalizationDescriptor& desc); }; -template <> struct StringifyLayerParameters<L2NormalizationDescriptor> +template <> struct StringifyLayerParameters<OriginsDescriptor> { - static void Serialize(ParameterStringifyFunction& fn, const L2NormalizationDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const OriginsDescriptor& desc); }; -template <> struct StringifyLayerParameters<BatchToSpaceNdDescriptor> +template <> struct StringifyLayerParameters<PadDescriptor> { - static void Serialize(ParameterStringifyFunction& fn, const BatchToSpaceNdDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const PadDescriptor& desc); +}; +template <> struct StringifyLayerParameters<PermuteDescriptor> +{ + static void Serialize(ParameterStringifyFunction& fn, const PermuteDescriptor& desc); }; -template <> struct StringifyLayerParameters<FakeQuantizationDescriptor> +template <> struct StringifyLayerParameters<Pooling2dDescriptor> { - static void Serialize(ParameterStringifyFunction& fn, const FakeQuantizationDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const Pooling2dDescriptor& desc); }; -template <> struct StringifyLayerParameters<ResizeBilinearDescriptor> +template <> struct StringifyLayerParameters<PreCompiledDescriptor> { - static void Serialize(ParameterStringifyFunction& fn, const ResizeBilinearDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const PreCompiledDescriptor& desc); }; -template <> struct StringifyLayerParameters<ResizeDescriptor> +template <> struct StringifyLayerParameters<ReduceDescriptor> { - static void Serialize(ParameterStringifyFunction& fn, const ResizeDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const ReduceDescriptor& desc); }; -template <> struct StringifyLayerParameters<SpaceToBatchNdDescriptor> +template <> struct StringifyLayerParameters<ReshapeDescriptor> { - static void Serialize(ParameterStringifyFunction& fn, const SpaceToBatchNdDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const ReshapeDescriptor& desc); }; -template <> struct StringifyLayerParameters<SpaceToDepthDescriptor> +template <> struct StringifyLayerParameters<ResizeBilinearDescriptor> { - static void Serialize(ParameterStringifyFunction& fn, const SpaceToDepthDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const ResizeBilinearDescriptor& desc); }; -template <> struct StringifyLayerParameters<LstmDescriptor> +template <> struct StringifyLayerParameters<ResizeDescriptor> { - static void Serialize(ParameterStringifyFunction& fn, const LstmDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const ResizeDescriptor& desc); }; -template <> struct StringifyLayerParameters<MeanDescriptor> +template <> struct StringifyLayerParameters<SpaceToBatchNdDescriptor> { - static void Serialize(ParameterStringifyFunction& fn, const MeanDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const SpaceToBatchNdDescriptor& desc); }; -template <> struct StringifyLayerParameters<PadDescriptor> +template <> struct StringifyLayerParameters<SpaceToDepthDescriptor> { - static void Serialize(ParameterStringifyFunction& fn, const PadDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const SpaceToDepthDescriptor& desc); }; template <> struct StringifyLayerParameters<StackDescriptor> @@ -160,9 +159,9 @@ template <> struct StringifyLayerParameters<StridedSliceDescriptor> static void Serialize(ParameterStringifyFunction& fn, const StridedSliceDescriptor& desc); }; -template <> struct StringifyLayerParameters<PreCompiledDescriptor> +template <> struct StringifyLayerParameters<SoftmaxDescriptor> { - static void Serialize(ParameterStringifyFunction& fn, const PreCompiledDescriptor& desc); + static void Serialize(ParameterStringifyFunction& fn, const SoftmaxDescriptor& desc); }; template <> struct StringifyLayerParameters<TransposeConvolution2dDescriptor> @@ -175,4 +174,9 @@ template <> struct StringifyLayerParameters<TransposeDescriptor> static void Serialize(ParameterStringifyFunction& fn, const TransposeDescriptor& desc); }; +template <> struct StringifyLayerParameters<ViewsDescriptor> +{ + static void Serialize(ParameterStringifyFunction& fn, const ViewsDescriptor& desc); +}; + } // namespace armnn
\ No newline at end of file |