diff options
| author | Francis Murtagh <francis.murtagh@arm.com> | 2022-06-27 12:44:50 +0100 |
|---|---|---|
| committer | Nikhil Raj <nikhil.raj@arm.com> | 2022-07-08 15:21:39 +0100 |
| commit | b619b42118f54c2db1946aef46477926efa4b75c (patch) | |
| tree | 00a101ce4a4db89481efe36ef312e6a1aae78126 /include/armnn/backends/SubgraphView.hpp | |
| parent | f4ccb1f6339a1e9ed573f188e7f14353167b5749 (diff) | |
| download | armnn-b619b42118f54c2db1946aef46477926efa4b75c.tar.gz | |
IVGCVSW-7034 Modified SubgraphView returned by GetWorkingCopy()
* Add virtual GetSlotIndex to IInputSlot
* Fix logic in GetWorkingCopy to use index of slots; so as not
to add slots to cloned subgraphView if not in original subgraphView
* Add test to cover cases when not all inputSlots to subgraphView layer
are part of the original subgraphView
* Mark SubgraphView::GetWorkingCopy() as const
Change-Id: I1d540f84c57f97f6c834ec06ca13393ffa55d379
Diffstat (limited to 'include/armnn/backends/SubgraphView.hpp')
| -rw-r--r-- | include/armnn/backends/SubgraphView.hpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/include/armnn/backends/SubgraphView.hpp b/include/armnn/backends/SubgraphView.hpp index dbf0544651..72d36829f4 100644 --- a/include/armnn/backends/SubgraphView.hpp +++ b/include/armnn/backends/SubgraphView.hpp @@ -156,7 +156,7 @@ public: /// Backend users should edit this working copy and then add it as a SubstitutionPair, along with original /// SubgraphView, to the OptimizationViews returned by OptimizeSubgraphView. /// ArmNN will then decide on whether or not to carry out Substitution of the two SubgraphViews. - SubgraphView GetWorkingCopy(); + SubgraphView GetWorkingCopy() const; /// These methods should be called on a working copy subgraph created from GetWorkingCopy. /// They take a SubgraphView pattern to replace and the substitute layer or subgraphView to substitute in. |