ArmNN
 25.11
Loading...
Searching...
No Matches
SubgraphUtils.hpp File Reference
#include <armnn/StrategyBase.hpp>
#include <armnn/Descriptors.hpp>
#include <optimizations/FoldPadIntoLayer2d.hpp>
#include <type_traits>
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.

Classes

struct  IsValidPair< L, D >
struct  IsValidPair< Pooling2dLayer, Pooling2dDescriptor >
struct  IsValidPair< Convolution2dLayer, Convolution2dDescriptor >
struct  IsValidPair< DepthwiseConvolution2dLayer, DepthwiseConvolution2dDescriptor >
class  OpBlockSequencer< LayerT, BlockT >
struct  OpBlockSequencer< LayerT, BlockT >::Pair

Namespaces

namespace  armnn
 Copyright (c) 2021 ARM Limited and Contributors.
namespace  armnn::FoldPadConstraints

Functions

bool IsNCHW (armnn::Layer &layer)
void ReportUntouchedLayers (OptimizationViews &optimizationViews, std::map< LayerGuid, Layer * > untouched)
template<typename LayerType>
LayerTypeReplaceLayer (OptimizationViews &optimizationViews, LayerType *baseLayer, LayerType *replacementLayer)
template<typename LayerType>
LayerTypeFoldPadLayer (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.
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.
void RemoveReshapeLayer (ReshapeLayer *baseLayer, std::map< LayerGuid, Layer * > &untouched, OptimizationViews &optimizationViews)
template<typename LayerT, typename Descriptor>
void FoldPadLayer2d (OptimizationViews &optimizationViews, LayerT *baseLayer, Descriptor &descriptor, 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)

Variables

template<class>
constexpr bool alwaysFalse = false