diff options
author | Jim Flynn <jim.flynn@arm.com> | 2019-12-12 22:09:51 +0000 |
---|---|---|
committer | Jim Flynn Arm <jim.flynn@arm.com> | 2019-12-18 11:11:18 +0000 |
commit | 082758c14ef0cddfb9cc87716f706d5917ddb62f (patch) | |
tree | 0a067822666191432fa5fabb7533ac042779ae66 /src/armnn/SerializeLayerParameters.hpp | |
parent | 377c806225a8d6506ff6e54a64949af039ea79a3 (diff) | |
download | armnn-082758c14ef0cddfb9cc87716f706d5917ddb62f.tar.gz |
IVGCVSW-4268 Print all Descriptors on dot graph
Change-Id: Ibc174f244bc72ff928879b0ea32f4f2c51f3a3b7
Signed-off-by: Jim Flynn <jim.flynn@arm.com>
Signed-off-by: Kevin May <kevin.may@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 |