aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/Layers.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/Layers.hpp')
-rw-r--r--src/armnn/Layers.hpp9
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,