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