diff options
| author | Derek Lamberti <derek.lamberti@arm.com> | 2020-12-07 13:54:12 +0000 |
|---|---|---|
| committer | Matthew Bentham <matthew.bentham@arm.com> | 2020-12-11 09:02:52 +0000 |
| commit | 161d29ca4091bb1ff79736a416978b49989fc622 (patch) | |
| tree | 971ccca037dc6ac5eb0d45d7981f1a3feb19e023 /src/armnn/SubgraphView.hpp | |
| parent | 00f9d77625cec8c4191c1a85dffd6ec2019efc8d (diff) | |
| download | armnn-161d29ca4091bb1ff79736a416978b49989fc622.tar.gz | |
Sort subgraphview layers on construction
Make it easier for backends to traverse the subgraph during optimization
Change-Id: I140cb11f78bab5f19c801a5b55efffb38c63837f
Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>
Diffstat (limited to 'src/armnn/SubgraphView.hpp')
| -rw-r--r-- | src/armnn/SubgraphView.hpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/armnn/SubgraphView.hpp b/src/armnn/SubgraphView.hpp index 1b9c308daa..cb9e415dd2 100644 --- a/src/armnn/SubgraphView.hpp +++ b/src/armnn/SubgraphView.hpp @@ -86,6 +86,9 @@ public: private: void CheckSubgraph(); + /// Arrange the order of layers topologically so that nodes can be visited in valid order + void ArrangeBySortOrder(); + /// The list of pointers to the input slots of the parent graph. InputSlots m_InputSlots; |