24.02
|
#include "SubgraphViewSelector.hpp"
#include "Graph.hpp"
#include <armnn/utility/Assert.hpp>
#include <armnn/utility/IgnoreUnused.hpp>
#include <armnn/utility/PolymorphicDowncast.hpp>
#include <algorithm>
#include <map>
#include <queue>
#include <unordered_set>
Go to the source code of this file.
Namespaces | |
armnn | |
Copyright (c) 2021 ARM Limited and Contributors. | |
Functions | |
template<typename Delegate > | |
void | ForEachLayerInput (LayerSelectionInfo::LayerInfoContainer &layerInfos, LayerSelectionInfo &layerInfo, Delegate function) |
template<typename Delegate > | |
void | ForEachLayerOutput (LayerSelectionInfo::LayerInfoContainer &layerInfos, LayerSelectionInfo &layerInfo, Delegate function) |
void | AssignSplitId (LayerSelectionInfo::LayerInfoContainer &layerInfos, LayerSelectionInfo &layerInfo) |
bool | IsReadyForSplitAssignment (LayerSelectionInfo::LayerInfoContainer &layerInfos, LayerSelectionInfo &layerInfo) |
bool m_IsProcessed |
Definition at line 253 of file SubgraphViewSelector.cpp.
bool m_IsSelected |
Definition at line 252 of file SubgraphViewSelector.cpp.
IConnectableLayer* m_Layer |
Definition at line 247 of file SubgraphViewSelector.cpp.
std::shared_ptr<PartialSubgraph> m_Subgraph |
Which subgraph this layer has been assigned to.
Only valid once m_IsProcessed is true. Two layers with different m_Subgraph pointers may in fact have been merged into the same subgraph - see the description of the PartialSubgraph class.
Definition at line 251 of file SubgraphViewSelector.cpp.