diff options
| author | Francis Murtagh <francis.murtagh@arm.com> | 2022-01-19 16:31:58 +0000 |
|---|---|---|
| committer | Francis Murtagh <francis.murtagh@arm.com> | 2022-02-03 17:34:45 +0000 |
| commit | 9d74ba6e85a043e9603445e062315f5c4965fbd6 (patch) | |
| tree | 1997aac65827e64d4270149826ba92c6fa3e4d11 /src/armnn/Graph.hpp | |
| parent | 2e24175c683bca42496104591d6b702dad360b8e (diff) | |
| download | armnn-9d74ba6e85a043e9603445e062315f5c4965fbd6.tar.gz | |
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 <francis.murtagh@arm.com>
Change-Id: Iaaef14448d8b73867eaee9d69f4f98d5d1bf171c
Diffstat (limited to 'src/armnn/Graph.hpp')
| -rw-r--r-- | src/armnn/Graph.hpp | 2 |
1 files changed, 2 insertions, 0 deletions
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. |