ArmNN
 24.02
FoldPadIntoLayer2d.hpp File Reference
Include dependency graph for FoldPadIntoLayer2d.hpp:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

class  FoldPadIntoConvolution2dImpl
 
class  FoldPadIntoDepthwiseConvolution2dImpl
 
class  FoldPadIntoPooling2dImpl
 

Namespaces

 armnn
 Copyright (c) 2021 ARM Limited and Contributors.
 
 armnn::optimizations
 
 armnn::optimizations::pad_fold
 

Typedefs

using FoldPadIntoConvolution2d = OptimizeForExclusiveConnection< PadLayer, Convolution2dLayer, pad_fold::FoldPadIntoConvolution2dImpl >
 
using FoldPadIntoDepthwiseConvolution2d = OptimizeForExclusiveConnection< PadLayer, DepthwiseConvolution2dLayer, pad_fold::FoldPadIntoDepthwiseConvolution2dImpl >
 
using FoldPadIntoPooling2d = OptimizeForExclusiveConnection< PadLayer, Pooling2dLayer, pad_fold::FoldPadIntoPooling2dImpl >
 

Functions

float GetZeroElement (const TensorInfo &tensorInfo)
 
float GetLowestElement (const TensorInfo &tensorInfo)
 
bool IsNeutralElement (const Convolution2dDescriptor &, const TensorInfo &tensorInfo, const float tensorValue)
 
bool IsNeutralElement (const DepthwiseConvolution2dDescriptor &, const TensorInfo &tensorInfo, const float tensorValue)
 
bool IsNeutralElement (const Pooling2dDescriptor &descriptor, const TensorInfo &tensorInfo, const float tensorValue)
 
bool IsPooling2dPadded (const Pooling2dDescriptor &poolDescriptor)
 
template<typename Descriptor >
bool TryFoldPadIntoLayer2d (const PadDescriptor &padDescriptor, Descriptor &layerDescriptor, const TensorInfo &tensorInfo)
 
bool TryFoldPadIntoLayer2d (const PadDescriptor &padDescriptor, Pooling2dDescriptor &poolDescriptor, const TensorInfo &tensorInfo, bool isBackendOptimization=false)
 
template<typename Layer2dT >
Layer2dT * FoldPadIntoLayer2dImpl (Graph &graph, InputSlot &connection)