diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/armnn/Layer.cpp | 5 | ||||
| -rw-r--r-- | src/backends/backendsCommon/test/OptimizedNetworkTests.cpp | 9 |
2 files changed, 11 insertions, 3 deletions
diff --git a/src/armnn/Layer.cpp b/src/armnn/Layer.cpp index 2256e48c91..c8d5a1690d 100644 --- a/src/armnn/Layer.cpp +++ b/src/armnn/Layer.cpp @@ -445,8 +445,13 @@ void Layer::VerifyShapeInferenceType(const TensorShape& outputShape, ShapeInfere void Layer::SerializeLayerParameters(ParameterStringifyFunction& fn) const { + std::string guid = std::to_string(m_Guid); std::string layerType = GetLayerTypeAsCString(m_Type); std::string backendId = std::string(m_BackendId); + if (!(guid.compare("") == 0) && !guid.empty()) + { + fn("Guid", guid); + } if(!(m_LayerName.compare("") == 0) && !m_LayerName.empty()) { fn("LayerName",m_LayerName); diff --git a/src/backends/backendsCommon/test/OptimizedNetworkTests.cpp b/src/backends/backendsCommon/test/OptimizedNetworkTests.cpp index c8986f5d4d..721dfb004c 100644 --- a/src/backends/backendsCommon/test/OptimizedNetworkTests.cpp +++ b/src/backends/backendsCommon/test/OptimizedNetworkTests.cpp @@ -50,9 +50,12 @@ BOOST_AUTO_TEST_CASE(SerializeToDot) "digraph Optimized {\n" " node [shape=\"record\"];\n" " edge [fontsize=8 fontcolor=\"blue\" fontname=\"arial-bold\"];\n" - " " << inputId << " [label=\"{Input|LayerType : Input\\lBackendID : CpuRef\\l}\"];\n" - " " << addId << " [label=\"{Addition|LayerType : Addition\\lBackendID : CpuRef\\l}\"];\n" - " " << outputId << " [label=\"{Output|LayerType : Output\\lBackendID : CpuRef\\l}\"];\n" + " " << inputId << " [label=\"{Input|Guid : " << inputId << "\\lLayerType : Input\\l" + "BackendID : CpuRef\\l}\"];\n" + " " << addId << " [label=\"{Addition|Guid : " << addId << "\\lLayerType : Addition\\l" + "BackendID : CpuRef\\l}\"];\n" + " " << outputId << " [label=\"{Output|Guid : " << outputId << "\\lLayerType : Output\\l" + "BackendID : CpuRef\\l}\"];\n" " " << inputId << " -> " << addId << " [label=< [4] >];\n" " " << inputId << " -> " << addId << " [label=< [4] >];\n" " " << addId << " -> " << outputId << " [label=< [4] >];\n" |