diff options
author | Francis Murtagh <francis.murtagh@arm.com> | 2022-12-21 09:10:04 +0000 |
---|---|---|
committer | Francis Murtagh <francis.murtagh@arm.com> | 2023-01-12 11:11:28 +0000 |
commit | 01f72693d39ed966ad06adadc8aac141bc395659 (patch) | |
tree | d619d2c112bff714d38025b57370aa4095935e36 /src/armnn/SubgraphViewSelector.hpp | |
parent | 05b6a3e5946a3f58b6f5b0caface9153a1c2b364 (diff) | |
download | armnn-01f72693d39ed966ad06adadc8aac141bc395659.tar.gz |
IVGCVSW-7418 Allow working copy SubgraphView to get Original Slots
* API to remove need for workaround so backend users can get slots
* OutputSlots outside the SubgraphView needed to obtain TensorInfo
* Fix a few Copyright headers
* Add shared_ptr back to original subgraph view using
std::enable_shared_from_this
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>
Change-Id: I033a00d6fc4020619d406ac06a156b7e380a426a
Diffstat (limited to 'src/armnn/SubgraphViewSelector.hpp')
-rw-r--r-- | src/armnn/SubgraphViewSelector.hpp | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/src/armnn/SubgraphViewSelector.hpp b/src/armnn/SubgraphViewSelector.hpp index 0a05bc259e..4808a99fb9 100644 --- a/src/armnn/SubgraphViewSelector.hpp +++ b/src/armnn/SubgraphViewSelector.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2017 Arm Ltd. All rights reserved. +// Copyright © 2017, 2023 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once @@ -21,8 +21,7 @@ class Graph; class SubgraphViewSelector final { public: - using SubgraphViewPtr = std::unique_ptr<SubgraphView>; - using Subgraphs = std::vector<SubgraphViewPtr>; + using Subgraphs = std::vector<SubgraphView::SubgraphViewPtr>; using LayerSelectorFunction = std::function<bool(const Layer&)>; /// Selects subgraphs from a graph based on the selector function and the algorithm. |