FoldPadIntoLayer2d.hpp File Reference

#include "Optimization.hpp"
#include <armnnUtils/QuantizeHelper.hpp>
#include <armnn/utility/PolymorphicDowncast.hpp>
#include <armnnUtils/DataLayoutIndexed.hpp>

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)