diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2020-01-23 11:44:24 +0000 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2020-01-23 12:04:19 +0000 |
commit | 190a39a4a9598e42b636ae4ab843761884148160 (patch) | |
tree | 84f9271361f93b15c7fbecc5ba3674f5e67a0cd8 /src/armnn/SerializeLayerParameters.hpp | |
parent | 5277ad5a23212f29aa50faa83521419722d2d949 (diff) | |
download | armnn-190a39a4a9598e42b636ae4ab843761884148160.tar.gz |
IVGCVSW-4268 Print all Descriptors on dot graph
Change-Id: I82a7dcffc3771efa8350e2e1e22cfb20969551f9
Signed-off-by: Jim Flynn <jim.flynn@arm.com>
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Diffstat (limited to 'src/armnn/SerializeLayerParameters.hpp')
-rw-r--r-- | src/armnn/SerializeLayerParameters.hpp | 111 |
1 files changed, 98 insertions, 13 deletions
diff --git a/src/armnn/SerializeLayerParameters.hpp b/src/armnn/SerializeLayerParameters.hpp index a47ce5a970..76ca0a52dd 100644 --- a/src/armnn/SerializeLayerParameters.hpp +++ b/src/armnn/SerializeLayerParameters.hpp @@ -11,7 +11,7 @@ namespace armnn { -using ParameterStringifyFunction = std::function<void(const std::string & name, const std::string & value)>; +using ParameterStringifyFunction = std::function<void(const std::string& name, const std::string& value)>; /// /// StringifyLayerParameters allows serializing layer parameters to string. @@ -22,57 +22,142 @@ using ParameterStringifyFunction = std::function<void(const std::string & name, template <typename LayerParameter> struct StringifyLayerParameters { - static void Serialize(ParameterStringifyFunction &, const LayerParameter &) {} + static void Serialize(ParameterStringifyFunction&, const LayerParameter&) {} }; template <> struct StringifyLayerParameters<PermuteDescriptor> { - static void Serialize(ParameterStringifyFunction & fn, const PermuteDescriptor & desc); + static void Serialize(ParameterStringifyFunction& fn, const PermuteDescriptor& desc); }; template <> struct StringifyLayerParameters<ReshapeDescriptor> { - static void Serialize(ParameterStringifyFunction & fn, const ReshapeDescriptor & desc); + static void Serialize(ParameterStringifyFunction& fn, const ReshapeDescriptor& desc); }; template <> struct StringifyLayerParameters<ActivationDescriptor> { - static void Serialize(ParameterStringifyFunction & fn, const ActivationDescriptor & desc); + static void Serialize(ParameterStringifyFunction& fn, const ActivationDescriptor& desc); }; template <> struct StringifyLayerParameters<Convolution2dDescriptor> { - static void Serialize(ParameterStringifyFunction & fn, const Convolution2dDescriptor & desc); + static void Serialize(ParameterStringifyFunction& fn, const Convolution2dDescriptor& desc); }; template <> struct StringifyLayerParameters<BatchNormalizationDescriptor> { - static void Serialize(ParameterStringifyFunction & fn, const BatchNormalizationDescriptor & desc); + static void Serialize(ParameterStringifyFunction& fn, const BatchNormalizationDescriptor& desc); }; template <> struct StringifyLayerParameters<DepthwiseConvolution2dDescriptor> { - static void Serialize(ParameterStringifyFunction & fn, const DepthwiseConvolution2dDescriptor & desc); + static void Serialize(ParameterStringifyFunction& fn, const DepthwiseConvolution2dDescriptor& desc); }; template <> struct StringifyLayerParameters<Pooling2dDescriptor> { - static void Serialize(ParameterStringifyFunction & fn, const Pooling2dDescriptor & desc); + static void Serialize(ParameterStringifyFunction& fn, const Pooling2dDescriptor& desc); }; template <> struct StringifyLayerParameters<SoftmaxDescriptor> { - static void Serialize(ParameterStringifyFunction & fn, const SoftmaxDescriptor & desc); + static void Serialize(ParameterStringifyFunction& fn, const SoftmaxDescriptor& desc); }; template <> struct StringifyLayerParameters<FullyConnectedDescriptor> { - static void Serialize(ParameterStringifyFunction & fn, const FullyConnectedDescriptor & desc); + static void Serialize(ParameterStringifyFunction& fn, const FullyConnectedDescriptor& desc); }; template <> struct StringifyLayerParameters<OriginsDescriptor> { - static void Serialize(ParameterStringifyFunction & fn, const OriginsDescriptor & desc); + static void Serialize(ParameterStringifyFunction& fn, const OriginsDescriptor& desc); }; -}
\ No newline at end of file +template <> struct StringifyLayerParameters<ViewsDescriptor> +{ + static void Serialize(ParameterStringifyFunction& fn, const ViewsDescriptor& desc); +}; + +template <> struct StringifyLayerParameters<DetectionPostProcessDescriptor> +{ + static void Serialize(ParameterStringifyFunction& fn, const DetectionPostProcessDescriptor& desc); +}; + +template <> struct StringifyLayerParameters<NormalizationDescriptor> +{ + static void Serialize(ParameterStringifyFunction& fn, const NormalizationDescriptor& desc); +}; + +template <> struct StringifyLayerParameters<L2NormalizationDescriptor> +{ + static void Serialize(ParameterStringifyFunction& fn, const L2NormalizationDescriptor& desc); +}; + +template <> struct StringifyLayerParameters<BatchToSpaceNdDescriptor> +{ + static void Serialize(ParameterStringifyFunction& fn, const BatchToSpaceNdDescriptor& desc); +}; + +template <> struct StringifyLayerParameters<FakeQuantizationDescriptor> +{ + static void Serialize(ParameterStringifyFunction& fn, const FakeQuantizationDescriptor& desc); +}; + +template <> struct StringifyLayerParameters<ResizeBilinearDescriptor> +{ + static void Serialize(ParameterStringifyFunction& fn, const ResizeBilinearDescriptor& desc); +}; + +template <> struct StringifyLayerParameters<ResizeDescriptor> +{ + static void Serialize(ParameterStringifyFunction& fn, const ResizeDescriptor& desc); +}; + +template <> struct StringifyLayerParameters<SpaceToBatchNdDescriptor> +{ + static void Serialize(ParameterStringifyFunction& fn, const SpaceToBatchNdDescriptor& desc); +}; + +template <> struct StringifyLayerParameters<SpaceToDepthDescriptor> +{ + static void Serialize(ParameterStringifyFunction& fn, const SpaceToDepthDescriptor& desc); +}; + +template <> struct StringifyLayerParameters<LstmDescriptor> +{ + static void Serialize(ParameterStringifyFunction& fn, const LstmDescriptor& desc); +}; + +template <> struct StringifyLayerParameters<MeanDescriptor> +{ + static void Serialize(ParameterStringifyFunction& fn, const MeanDescriptor& desc); +}; + +template <> struct StringifyLayerParameters<PadDescriptor> +{ + static void Serialize(ParameterStringifyFunction& fn, const PadDescriptor& desc); +}; + +template <> struct StringifyLayerParameters<StackDescriptor> +{ + static void Serialize(ParameterStringifyFunction& fn, const StackDescriptor& desc); +}; + +template <> struct StringifyLayerParameters<StridedSliceDescriptor> +{ + static void Serialize(ParameterStringifyFunction& fn, const StridedSliceDescriptor& desc); +}; + +template <> struct StringifyLayerParameters<PreCompiledDescriptor> +{ + static void Serialize(ParameterStringifyFunction& fn, const PreCompiledDescriptor& desc); +}; + +template <> struct StringifyLayerParameters<TransposeConvolution2dDescriptor> +{ + static void Serialize(ParameterStringifyFunction& fn, const TransposeConvolution2dDescriptor& desc); +}; + +} // namespace armnn
\ No newline at end of file |