diff options
Diffstat (limited to 'src/armnn/Layers.hpp')
-rw-r--r-- | src/armnn/Layers.hpp | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/src/armnn/Layers.hpp b/src/armnn/Layers.hpp index 5a1e3ca063..cb460e125f 100644 --- a/src/armnn/Layers.hpp +++ b/src/armnn/Layers.hpp @@ -22,10 +22,17 @@ template <typename Parameters> class LayerWithParameters : public Layer { public: - typedef Parameters DescriptorType; + using DescriptorType = Parameters; const Parameters& GetParameters() const { return m_Param; } + /// Helper to serialize the layer parameters to string + /// (currently used in DotSerializer and company) + void SerializeLayerParameters(ParameterStringifyFunction & fn) const + { + StringifyLayerParameters<Parameters>::Serialize(fn, m_Param); + } + protected: LayerWithParameters(unsigned int numInputSlots, unsigned int numOutputSlots, |