diff options
author | Francis Murtagh <francis.murtagh@arm.com> | 2022-06-27 12:44:50 +0100 |
---|---|---|
committer | Colm Donelan <colm.donelan@arm.com> | 2022-07-06 08:54:39 +0000 |
commit | cea3d49619a87ffb81422c7e9383368baa93a408 (patch) | |
tree | 00a101ce4a4db89481efe36ef312e6a1aae78126 /include/armnn/backends | |
parent | 1a7f033768acb27da11503bd29abb468d2e77f9e (diff) | |
download | armnn-cea3d49619a87ffb81422c7e9383368baa93a408.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')
-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. |