#include <FoldPadIntoLayer2d.hpp>

Public Member Functions
void	Run (Graph &graph, InputSlot &connection) const

Protected Member Functions
	FoldPadIntoDepthwiseConvolution2dImpl ()=default

	~FoldPadIntoDepthwiseConvolution2dImpl ()=default

Detailed Description

Definition at line 189 of file FoldPadIntoLayer2d.hpp.

Constructor & Destructor Documentation

◆ FoldPadIntoDepthwiseConvolution2dImpl()

FoldPadIntoDepthwiseConvolution2dImpl ( )

protecteddefault

◆ ~FoldPadIntoDepthwiseConvolution2dImpl()

~FoldPadIntoDepthwiseConvolution2dImpl ( )

protecteddefault

Member Function Documentation

◆ Run()

void Run	(	Graph &	graph,
		InputSlot &	connection
	)		const

inline

Definition at line 192 of file FoldPadIntoLayer2d.hpp.

References ARMNN_ASSERT_MSG, and InputSlot::GetOwningLayer().

     {
         const auto newConv2dLayer = FoldPadIntoLayer2dImpl<DepthwiseConvolution2dLayer>(graph, connection);
 
         if (newConv2dLayer != nullptr)
         {
             const auto conv2dLayer = PolymorphicDowncast<DepthwiseConvolution2dLayer*>(&connection.GetOwningLayer());
             // Copy weights and bias to the new convolution layer
             ARMNN_ASSERT_MSG(conv2dLayer->m_Weight != nullptr,
                              "FoldPadIntoDepthwiseConvolution2d: Weights data should not be null.");
             newConv2dLayer->m_Weight = std::move(conv2dLayer->m_Weight);
 
             if (conv2dLayer->GetParameters().m_BiasEnabled)
             {
                 ARMNN_ASSERT_MSG(conv2dLayer->m_Bias != nullptr,
                                 "FoldPadIntoDepthwiseConvolution2d: Bias data should not be null if bias is enabled.");
                 newConv2dLayer->m_Bias = std::move(conv2dLayer->m_Bias);
             }
         }
     }

The documentation for this class was generated from the following file:

src/armnn/optimizations/FoldPadIntoLayer2d.hpp