#include <FoldPadIntoLayer2d.hpp>
Definition at line 223 of file FoldPadIntoLayer2d.hpp.
◆ FoldPadIntoDepthwiseConvolution2dImpl()
◆ ~FoldPadIntoDepthwiseConvolution2dImpl()
◆ Run()
Definition at line 226 of file FoldPadIntoLayer2d.hpp.
References ARMNN_ASSERT_MSG, and InputSlot::GetOwningLayer().
228 const auto newConv2dLayer = FoldPadIntoLayer2dImpl<DepthwiseConvolution2dLayer>(graph, connection);
230 if (newConv2dLayer !=
nullptr)
232 const auto conv2dLayer = PolymorphicDowncast<DepthwiseConvolution2dLayer*>(&connection.GetOwningLayer());
234 ARMNN_ASSERT_MSG(newConv2dLayer->GetInputSlot(1).GetConnection() !=
nullptr,
235 "FoldPadIntoDepthwiseConvolution2d: New convolution layer is missing connection to weights layer");
238 newConv2dLayer->m_Weight = std::move(conv2dLayer->m_Weight);
240 if (conv2dLayer->GetParameters().m_BiasEnabled)
242 ARMNN_ASSERT_MSG(newConv2dLayer->GetInputSlot(2).GetConnection() !=
nullptr,
243 "FoldPadIntoConvolution2d: New convolution layer is missing " 244 "connection to bias layer.");
246 newConv2dLayer->m_Bias = std::move(conv2dLayer->m_Bias);
#define ARMNN_ASSERT_MSG(COND, MSG)
The documentation for this class was generated from the following file: