PermuteAndBatchToSpaceAsDepthToSpace.hpp File Reference

#include "Optimization.hpp"

Go to the source code of this file.

Classes
class	PermuteAndBatchToSpaceAsDepthToSpaceImpl< PermuteType >
	Replaces Permute leading into BatchToSpace with a DepthToSpace in the case where the Permute swaps the batch and channels dimensions such that the replacement is valid. More...

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

Typedefs
using	PermuteAndBatchToSpaceAsDepthToSpace = OptimizeForConnection< PermuteLayer, BatchToSpaceNdLayer, PermuteAndBatchToSpaceAsDepthToSpaceImpl< PermuteLayer > >
using	TransposeAndBatchToSpaceAsDepthToSpace = OptimizeForConnection< TransposeLayer, BatchToSpaceNdLayer, PermuteAndBatchToSpaceAsDepthToSpaceImpl< TransposeLayer > >