ArmNN
 24.08
SubgraphUtils.hpp File Reference
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)
 
bool IsSequenceLayerType (Layer &layer, LayerType type)
 
bool IsSequenceLayerType (Layer &layer, BinaryOperation type)
 
template<typename TYPE >
bool IsLayerSequence (Layer &currentLayer, TYPE first, TYPE second, TYPE third, Layer *layerList[4], bool handleValidActivates, const std::vector< ActivationFunction > &validActivates)