aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/Layer.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/Layer.cpp')
-rw-r--r--src/armnn/Layer.cpp19
1 files changed, 19 insertions, 0 deletions
diff --git a/src/armnn/Layer.cpp b/src/armnn/Layer.cpp
index 528020bab5..1e384336c9 100644
--- a/src/armnn/Layer.cpp
+++ b/src/armnn/Layer.cpp
@@ -400,4 +400,23 @@ std::vector<TensorShape> Layer::InferOutputShapes(const std::vector<TensorShape>
}
return inputShapes;
}
+
+void Layer::SerializeLayerParameters(ParameterStringifyFunction& fn) const
+{
+ std::string layerType = GetLayerTypeAsCString(m_Type);
+ std::string backendId = std::string(m_BackendId);
+ if(!(m_LayerName.compare("") == 0) && !m_LayerName.empty())
+ {
+ fn("LayerName",m_LayerName);
+ }
+ if(!(layerType.compare("") == 0) && !layerType.empty())
+ {
+ fn("LayerType",layerType);
+ }
+ if(!(backendId.compare("") == 0) && !backendId.empty())
+ {
+ fn("BackendID",backendId);
+ }
+}
+
} // namespace armnn