ArmNN
NotReleased
|
#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) |
bool m_IsProcessed |
Definition at line 245 of file SubgraphViewSelector.cpp.
bool m_IsSelected |
Definition at line 244 of file SubgraphViewSelector.cpp.
Layer* m_Layer |
Definition at line 239 of file SubgraphViewSelector.cpp.
Referenced by armnnTfParser::CalcPadding(), and armnnTfParser::OutputShapeOfSqueeze().
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.