From 9d74ba6e85a043e9603445e062315f5c4965fbd6 Mon Sep 17 00:00:00 2001 From: Francis Murtagh Date: Wed, 19 Jan 2022 16:31:58 +0000 Subject: IVGCVSW-6634 SubgraphView: Add method of returning a GetSubgraphWorkingCopy * Add pointer to SubgraphView allowing it to store a working copy implementation of its own representation of graph. * Make SubgraphView a friend of Graph to allow access to layers. * Add constructor to SubgraphView taking SubgraphViewWorkingCopyPtr * Rewrite Graph::SubstituteSubgraph for use on SubgraphView * Add GetWorkingCopy() method * Add tests for replacement of multiplication with DepthwiseConv2d * Check GetBackendHint() has value before passing to PrecompiledLayer * Add GetOwningIConnectableLayer to IInputSlot to allow traversing from IConnectableLayer->IOutputSlot->IInputSlot->IConnectableLayer Signed-off-by: Francis Murtagh Change-Id: Iaaef14448d8b73867eaee9d69f4f98d5d1bf171c --- src/armnn/Graph.hpp | 2 ++ 1 file changed, 2 insertions(+) (limited to 'src/armnn/Graph.hpp') diff --git a/src/armnn/Graph.hpp b/src/armnn/Graph.hpp index d71149d069..0c34d35685 100644 --- a/src/armnn/Graph.hpp +++ b/src/armnn/Graph.hpp @@ -280,6 +280,8 @@ private: /// Also verifies weights and bias are set for FullyConnected layers. void ConstructErrorMessageForUnconnectedInputs(Layer* const layer, unsigned int slotIndex); + + friend class SubgraphView; }; /// Common base class for layers in the graph. -- cgit v1.2.1