SubgraphUtils.hpp File Reference

#include <armnn/StrategyBase.hpp>
#include <armnn/Descriptors.hpp>
#include <optimizations/FoldPadIntoLayer2d.hpp>

Include dependency graph for SubgraphUtils.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces
	armnn
	Copyright (c) 2021 ARM Limited and Contributors.

Functions
bool	IsNCHW (armnn::Layer &layer)
void	ReportUntouchedLayers (OptimizationViews &optimizationViews, std::map< LayerGuid, Layer * > untouched)
template<typename LayerType >
LayerType *	FoldPadLayer (OptimizationViews &optimizationViews, LayerType *baseLayer, LayerType *replacementLayer, PadLayer *padLayer)
bool	ConnectedToLayerWithNCHW (Layer *baseLayer)
	Checks if the Layer is connected to any Layer that has an NCHW layout. More...
bool	ConnectedToLayerType (Layer *baseLayer, LayerType layerType, unsigned int dimSize=0)
	Checks the Layer's Connections to see if it's connected to a Layer with the provided layerType. More...
void	RemoveReshapeLayer (ReshapeLayer *baseLayer, std::map< LayerGuid, Layer * > &untouched, OptimizationViews &optimizationViews)
template<typename LayerType >
LayerType *	FoldPadIntoAveragePool2d (OptimizationViews &optimizationViews, Pooling2dLayer *baseLayer, Pooling2dDescriptor &poolDescriptor, PadLayer *padLayer)