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>

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 249 of file SubgraphViewSelector.cpp.

m_IsSelected

bool m_IsSelected

Definition at line 248 of file SubgraphViewSelector.cpp.

m_Layer

Layer* m_Layer

Definition at line 243 of file SubgraphViewSelector.cpp.

Referenced by armnnTfParser::CalcPadding(), ITfParser::TfParserImpl::ParseLrn(), and ITfParser::TfParserImpl::ParseMean().

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 247 of file SubgraphViewSelector.cpp.