ArmNN
NotReleased
|
#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 17 of file PermuteAndBatchToSpaceAsDepthToSpace.hpp.
Definition at line 11 of file PermuteAndBatchToSpaceAsDepthToSpace.cpp.
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, and OutputHandler::SetTensorInfo().