diff options
-rw-r--r-- | src/armnn/NetworkUtils.cpp | 6 | ||||
-rw-r--r-- | src/backends/reference/test/RefOptimizedNetworkTests.cpp | 4 |
2 files changed, 7 insertions, 3 deletions
diff --git a/src/armnn/NetworkUtils.cpp b/src/armnn/NetworkUtils.cpp index 285da4c9a9..666ce3d069 100644 --- a/src/armnn/NetworkUtils.cpp +++ b/src/armnn/NetworkUtils.cpp @@ -237,9 +237,11 @@ std::vector<DebugLayer*> InsertDebugLayerAfter(Graph& graph, Layer& layer) debugLayers.reserve(layer.GetNumOutputSlots()); // Connect a DebugLayer to each output slot of the layer + uint32_t outputSlotIdx = 0; for (auto outputSlot = layer.BeginOutputSlots(); outputSlot != layer.EndOutputSlots(); ++outputSlot) { - const std::string debugName = std::string("DebugLayerAfter") + layer.GetNameStr(); + const std::string debugName = std::string("DebugLayerAfter") + layer.GetNameStr() + "_" + + std::to_string(outputSlotIdx); DebugLayer* debugLayer = graph.InsertNewLayer<DebugLayer>(*outputSlot, debugName.c_str()); @@ -254,6 +256,8 @@ std::vector<DebugLayer*> InsertDebugLayerAfter(Graph& graph, Layer& layer) debugLayer->SetBackendId(Compute::CpuRef); debugLayers.emplace_back(debugLayer); + + ++outputSlotIdx; } return debugLayers; diff --git a/src/backends/reference/test/RefOptimizedNetworkTests.cpp b/src/backends/reference/test/RefOptimizedNetworkTests.cpp index 086c1e471a..2f25b6cd4d 100644 --- a/src/backends/reference/test/RefOptimizedNetworkTests.cpp +++ b/src/backends/reference/test/RefOptimizedNetworkTests.cpp @@ -196,9 +196,9 @@ BOOST_AUTO_TEST_CASE(DebugTestOnCpuRef) // Tests that the vertices exist and have correct names. BOOST_TEST(GraphHasNamedLayer(graph, "InputLayer")); - BOOST_TEST(GraphHasNamedLayer(graph, "DebugLayerAfterInputLayer")); + BOOST_TEST(GraphHasNamedLayer(graph, "DebugLayerAfterInputLayer_0")); BOOST_TEST(GraphHasNamedLayer(graph, "ActivationLayer")); - BOOST_TEST(GraphHasNamedLayer(graph, "DebugLayerAfterActivationLayer")); + BOOST_TEST(GraphHasNamedLayer(graph, "DebugLayerAfterActivationLayer_0")); BOOST_TEST(GraphHasNamedLayer(graph, "OutputLayer")); } |