IVGCVBENCH-1337 Added additional layer parameters to dot file and -v option

* Generic layer parameters now show up in dot file
* Convolution layer parameters have also been added to dot file
* ExecucteNetwork has an additional -v flag which generated dot file if there

Change-Id: I210bb19b45384eb3639b7e488c7a89049fa6f18d
Signed-off-by: Andre Ghattas <andre.ghattas@arm.com>
Signed-off-by: Szilard Papp <szilard.papp@arm.com>

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