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

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 > >