ArmNN
 24.02
SubgraphViewSelector.cpp File Reference
#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>
Include dependency graph for SubgraphViewSelector.cpp:

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)
 

Variable Documentation

◆ m_IsProcessed

bool m_IsProcessed

Definition at line 253 of file SubgraphViewSelector.cpp.

◆ m_IsSelected

bool m_IsSelected

Definition at line 252 of file SubgraphViewSelector.cpp.

◆ m_Layer

IConnectableLayer* m_Layer

Definition at line 247 of file SubgraphViewSelector.cpp.

◆ m_Subgraph

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.