#include <FoldPadIntoLayer2d.hpp>
Definition at line 208 of file FoldPadIntoLayer2d.hpp.
◆ FoldPadIntoConvolution2dImpl()
◆ ~FoldPadIntoConvolution2dImpl()
◆ Run()
Definition at line 211 of file FoldPadIntoLayer2d.hpp.
213 const auto newConv2dLayer = FoldPadIntoLayer2dImpl<Convolution2dLayer>(graph, connection);
215 if (newConv2dLayer !=
nullptr)
217 const auto conv2dLayer = PolymorphicDowncast<Convolution2dLayer*>(&connection.GetOwningLayer());
218 ARMNN_ASSERT_MSG(newConv2dLayer->GetInputSlot(1).GetConnection() !=
nullptr,
219 "FoldPadIntoConvolution2d: New convolution layer is missing connection to weights layer");
221 if (conv2dLayer->GetParameters().m_BiasEnabled)
223 ARMNN_ASSERT_MSG(newConv2dLayer->GetInputSlot(2).GetConnection() !=
nullptr,
224 "FoldPadIntoConvolution2d: New convolution layer is missing "
225 "connection to bias layer.");
References ARMNN_ASSERT_MSG, and InputSlot::GetOwningLayer().
The documentation for this class was generated from the following file: