FoldPadIntoConvolution2dImpl Class Reference

#include <FoldPadIntoLayer2d.hpp>

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

Protected Member Functions
	FoldPadIntoConvolution2dImpl ()=default
	~FoldPadIntoConvolution2dImpl ()=default

Detailed Description

Definition at line 160 of file FoldPadIntoLayer2d.hpp.

Constructor & Destructor Documentation

FoldPadIntoConvolution2dImpl()

FoldPadIntoConvolution2dImpl ( )

protecteddefault

Referenced by FoldPadIntoConvolution2dImpl::Run().

~FoldPadIntoConvolution2dImpl()

~FoldPadIntoConvolution2dImpl ( )

protecteddefault

Referenced by FoldPadIntoConvolution2dImpl::Run().

Member Function Documentation

Run()

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

inline

Definition at line 163 of file FoldPadIntoLayer2d.hpp.

References ARMNN_ASSERT_MSG, FoldPadIntoConvolution2dImpl::FoldPadIntoConvolution2dImpl(), InputSlot::GetOwningLayer(), and FoldPadIntoConvolution2dImpl::~FoldPadIntoConvolution2dImpl().

    {
        const auto newConv2dLayer = FoldPadIntoLayer2dImpl<Convolution2dLayer>(graph, connection);

        if (newConv2dLayer != nullptr)
        {
            const auto conv2dLayer = PolymorphicDowncast<Convolution2dLayer*>(&connection.GetOwningLayer());
            // Copy weights and bias to the new convolution layer
            ARMNN_ASSERT_MSG(conv2dLayer->m_Weight != nullptr,
                             "FoldPadIntoConvolution2d: 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,
                                 "FoldPadIntoConvolution2d: 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