UtilsGpuFsa.hpp File Reference

#include "armnn/Descriptors.hpp"
#include "arm_compute/dynamic_fusion/sketch/attributes/Conv2dAttributes.h"
#include "arm_compute/dynamic_fusion/sketch/attributes/DepthwiseConv2dAttributes.h"
#include "arm_compute/dynamic_fusion/sketch/attributes/Pool2dAttributes.h"
#include "arm_compute/dynamic_fusion/sketch/attributes/ResizeAttributes.h"

Include dependency graph for UtilsGpuFsa.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Functions
arm_compute::experimental::dynamic_fusion::Conv2dAttributes	CreateConv2dAttributes (const armnn::Convolution2dDescriptor &descriptor)
	Utility function used to setup an arm_compute::Conv2dAttributes object from given descriptor. More...
arm_compute::experimental::dynamic_fusion::DepthwiseConv2dAttributes	CreateDWConv2dAttributes (const armnn::DepthwiseConvolution2dDescriptor &descriptor, const unsigned int aclDepthMultiplier)
	Utility function used to setup an arm_compute::DepthwiseConv2dAttributes object from given descriptor. More...
arm_compute::experimental::dynamic_fusion::Pool2dAttributes	CreatePool2dAttributes (const armnn::Pooling2dDescriptor &descriptor)
	Utility function used to setup an arm_compute::Pool2dAttributes object from given descriptor. More...
arm_compute::experimental::dynamic_fusion::ResizeAttributes	CreateResizeAttributes (const armnn::ResizeDescriptor &descriptor)
	Utility function used to setup an arm_compute::ResizeDescriptor object from given descriptor. More...

Function Documentation

CreateConv2dAttributes()

arm_compute::experimental::dynamic_fusion::Conv2dAttributes CreateConv2dAttributes

(

const armnn::Convolution2dDescriptor &

descriptor

)

Utility function used to setup an arm_compute::Conv2dAttributes object from given descriptor.

Parameters

[in]

armnn::Convolution2dDescriptor

Returns

arm_compute::experimental::dynamic_fusion::Conv2dAttributes

Definition at line 14 of file UtilsGpuFsa.cpp.

{
    const arm_compute::Padding2D padInfo      = BuildArmComputePaddingInfo(descriptor);
    const arm_compute::Size2D    strideInfo   = BuildArmComputeSize2D(descriptor.m_StrideX, descriptor.m_StrideY);
    const arm_compute::Size2D    dilationInfo = BuildArmComputeSize2D(descriptor.m_DilationX, descriptor.m_DilationY);
 
    arm_compute::experimental::dynamic_fusion::Conv2dAttributes conv2dAttributes{};
    conv2dAttributes.pad(padInfo);
    conv2dAttributes.stride(strideInfo);
    conv2dAttributes.dilation(dilationInfo);
 
    return conv2dAttributes;
}

CreateDWConv2dAttributes()

arm_compute::experimental::dynamic_fusion::DepthwiseConv2dAttributes CreateDWConv2dAttributes	(	const armnn::DepthwiseConvolution2dDescriptor &	descriptor,
		const unsigned int	aclDepthMultiplier
	)

Utility function used to setup an arm_compute::DepthwiseConv2dAttributes object from given descriptor.

Parameters

[in]

armnn::DepthwiseConvolution2dDescriptor

Returns

arm_compute::experimental::dynamic_fusion::DepthwiseConv2dAttributes

Definition at line 29 of file UtilsGpuFsa.cpp.

{
    const arm_compute::Padding2D padInfo      = BuildArmComputePaddingInfo(descriptor);
    const arm_compute::Size2D    strideInfo   = BuildArmComputeSize2D(descriptor.m_StrideX, descriptor.m_StrideY);
    const arm_compute::Size2D    dilationInfo = BuildArmComputeSize2D(descriptor.m_DilationX, descriptor.m_DilationY);
 
    arm_compute::experimental::dynamic_fusion::DepthwiseConv2dAttributes depthwiseConv2dAttributes{};
    depthwiseConv2dAttributes.pad(padInfo);
    depthwiseConv2dAttributes.stride(strideInfo);
    depthwiseConv2dAttributes.dilation(dilationInfo);
    depthwiseConv2dAttributes.depth_multiplier(aclDepthMultiplier);
 
    return depthwiseConv2dAttributes;
}

CreatePool2dAttributes()

arm_compute::experimental::dynamic_fusion::Pool2dAttributes CreatePool2dAttributes

(

const armnn::Pooling2dDescriptor &

descriptor

)

Utility function used to setup an arm_compute::Pool2dAttributes object from given descriptor.

Parameters

[in]

armnn::Pooling2dDescriptor

Returns

arm_compute::experimental::dynamic_fusion::Pool2dAttributes

Definition at line 45 of file UtilsGpuFsa.cpp.

{
    const arm_compute::PoolingType poolType = ConvertPoolingAlgorithmToAclPoolingType(descriptor.m_PoolType);
    const arm_compute::Padding2D   padding  = BuildArmComputePaddingInfo(descriptor);
    const arm_compute::Size2D      poolSize = BuildArmComputeSize2D(descriptor.m_PoolWidth, descriptor.m_PoolHeight);
    const arm_compute::Size2D      strides  = BuildArmComputeSize2D(descriptor.m_StrideX, descriptor.m_StrideY);
    const bool excludePadding = (descriptor.m_PaddingMethod == PaddingMethod::Exclude);
 
    arm_compute::experimental::dynamic_fusion::Pool2dAttributes pool2dAttributes{};
    pool2dAttributes.pool_type(poolType);
    pool2dAttributes.pad(padding);
    pool2dAttributes.pool_size(poolSize);
    pool2dAttributes.stride(strides);
    pool2dAttributes.exclude_padding(excludePadding);
 
    return pool2dAttributes;
}

References armnn::ConvertPoolingAlgorithmToAclPoolingType(), and Pooling2dDescriptor::m_PoolType.

CreateResizeAttributes()

arm_compute::experimental::dynamic_fusion::ResizeAttributes CreateResizeAttributes

(

const armnn::ResizeDescriptor &

descriptor

)

Utility function used to setup an arm_compute::ResizeDescriptor object from given descriptor.

Parameters

[in]

armnn::ResizeDescriptor

Returns

arm_compute::experimental::dynamic_fusion::ResizeAttributes

Definition at line 64 of file UtilsGpuFsa.cpp.

{
    arm_compute::experimental::dynamic_fusion::ResizeAttributes resizeAttributes{};
    resizeAttributes.output_width(static_cast<int32_t>(descriptor.m_TargetWidth));
    resizeAttributes.output_height(static_cast<int32_t>(descriptor.m_TargetHeight));
    resizeAttributes.interpolation_policy(descriptor.m_Method == ResizeMethod::Bilinear ?
                                          arm_compute::InterpolationPolicy::BILINEAR :
                                          arm_compute::InterpolationPolicy::NEAREST_NEIGHBOR);
    resizeAttributes.sampling_policy(descriptor.m_HalfPixelCenters ? arm_compute::SamplingPolicy::CENTER
                                                                   : arm_compute::SamplingPolicy::TOP_LEFT);
    resizeAttributes.align_corners(descriptor.m_AlignCorners);
 
    return resizeAttributes;
}

References armnn::Bilinear, ResizeDescriptor::m_AlignCorners, ResizeDescriptor::m_HalfPixelCenters, ResizeDescriptor::m_Method, ResizeDescriptor::m_TargetHeight, and ResizeDescriptor::m_TargetWidth.