PermuteDepthwiseConv2dWeightsImpl Class Reference

#include <PermuteDepthwiseConv2dWeights.hpp>

Public Member Functions
void	Run (Graph &graph, Layer &layer) const

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

Detailed Description

Definition at line 19 of file PermuteDepthwiseConv2dWeights.hpp.

Constructor & Destructor Documentation

PermuteDepthwiseConv2dWeightsImpl()

PermuteDepthwiseConv2dWeightsImpl ( )

protecteddefault

~PermuteDepthwiseConv2dWeightsImpl()

~PermuteDepthwiseConv2dWeightsImpl ( )

protecteddefault

References Layer::GetBackendId(), InputSlot::GetConnectedOutputSlot(), armnn::GetDataLayoutName(), Layer::GetInputSlot(), Layer::GetOutputSlot(), LayerWithParameters< Parameters >::GetParameters(), OutputSlot::GetTensorInfo(), Graph::InsertNewLayer(), DepthwiseConvolution2dDescriptor::m_DataLayout, armnn::NCHW, armnn::NHWC, armnnUtils::Permuted(), Layer::SetBackendId(), and OutputSlot::SetTensorInfo().

Member Function Documentation

Run()

void Run ( Graph & graph, Layer & layer ) const

inline

Definition at line 23 of file PermuteDepthwiseConv2dWeights.hpp.

    {
        if (layer.GetType() == LayerType::DepthwiseConvolution2d)
        {
            AddPermuteLayer(graph, PolymorphicDowncast<DepthwiseConvolution2dLayer*>(&layer));
        }
    }

References armnn::DepthwiseConvolution2d, Layer::GetType(), and armnn::PolymorphicDowncast().

---

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

• src/armnn/optimizations/PermuteDepthwiseConv2dWeights.hpp