FoldPadIntoLayer2d.hpp File Reference

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

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
namespace	armnn
	Copyright (c) 2021 ARM Limited and Contributors.
namespace	armnn::optimizations
namespace	armnn::optimizations::pad_fold

Typedefs
using	FoldPadIntoConvolution2d
using	FoldPadIntoDepthwiseConvolution2d
using	FoldPadIntoPooling2d

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)