diff options
author | Jim Flynn <jim.flynn@arm.com> | 2019-12-12 22:09:51 +0000 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2019-12-12 22:12:48 +0000 |
commit | adfcb7bc6d7b875211c820ee58b46fd443ecb39d (patch) | |
tree | 63b49ab72534dfaa75ccecdad39aeafdbfba3ea3 /src/armnn/SerializeLayerParameters.hpp | |
parent | 8ac5fc39b2556fd3ac0d0c43ebf55525bde97f58 (diff) | |
download | armnn-adfcb7bc6d7b875211c820ee58b46fd443ecb39d.tar.gz |
IVGCVSW-4268 Print all Descriptors on dot graph
Change-Id: I82a7dcffc3771efa8350e2e1e22cfb20969551f9
Signed-off-by: Jim Flynn <jim.flynn@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 |