20.02
|
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...
#include <PermuteAndBatchToSpaceAsDepthToSpace.hpp>
Public Member Functions | |
void | Run (Graph &graph, InputSlot &connection) const |
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.
Definition at line 18 of file PermuteAndBatchToSpaceAsDepthToSpace.hpp.
Definition at line 21 of file PermuteAndBatchToSpaceAsDepthToSpace.hpp.
References armnn::BatchToSpaceNd, InputSlot::GetConnectedOutputSlot(), InputSlot::GetConnection(), Layer::GetInputSlot(), Layer::GetName(), TensorInfo::GetNumDimensions(), Layer::GetOutputHandler(), Layer::GetOutputSlot(), InputSlot::GetOwningLayer(), OutputSlot::GetOwningLayer(), IOutputSlot::GetTensorInfo(), OutputSlot::GetTensorInfo(), Layer::GetType(), BatchToSpaceNdDescriptor::m_Crops, BatchToSpaceNdDescriptor::m_DataLayout, armnn::NHWC, armnn::Permute, OutputHandler::SetTensorInfo(), and armnn::Transpose.