SubgraphViewSelector.cpp File Reference

#include "SubgraphViewSelector.hpp"
#include "Graph.hpp"
#include <boost/assert.hpp>
#include <algorithm>
#include <map>
#include <queue>
#include <unordered_set>

Go to the source code of this file.

Namespaces
	armnn

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

m_IsSelected

bool m_IsSelected

Definition at line 244 of file SubgraphViewSelector.cpp.

m_Layer

Layer* m_Layer

Definition at line 239 of file SubgraphViewSelector.cpp.

Referenced by armnnTfParser::CalcPadding(), and armnnTfParser::OutputShapeOfSqueeze().

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