aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/SerializeLayerParameters.hpp
diff options
context:
space:
mode:
authorTeresa Charlin <teresa.charlinreyes@arm.com>2021-06-04 10:29:47 +0100
committerTeresa Charlin <teresa.charlinreyes@arm.com>2021-06-04 10:29:47 +0100
commit400c593e03af3cc9131058b3cb30b787586f69a3 (patch)
treec57a916ac4bc95816a33330984016c8187d46ce9 /src/armnn/SerializeLayerParameters.hpp
parent1b1950d909e886ad3cbc0be68107fae8e4951283 (diff)
downloadarmnn-400c593e03af3cc9131058b3cb30b787586f69a3.tar.gz
COMPMID-4548 Print Elementwise Descriptor on dot graph
*Order the functions alphabetically Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com> Change-Id: Ia19f2727733026aefab17ebce0cabb66863a7af7
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