blob: a1d96f0ec115ef668bbba0d749d08929b3698929 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
//
// Copyright © 2023-2024 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//
#include "UtilsGpuFsa.hpp"
#include "aclCommon/ArmComputeTensorUtils.hpp"
#include "aclCommon/ArmComputeUtils.hpp"
using namespace armnn;
using namespace armnn::armcomputetensorutils;
using namespace arm_compute::experimental::dynamic_fusion;
Conv2dAttributes CreateConv2dAttributes(const Convolution2dDescriptor& descriptor)
{
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;
}
arm_compute::experimental::dynamic_fusion::DepthwiseConv2dAttributes
CreateDWConv2dAttributes(const DepthwiseConvolution2dDescriptor& descriptor, const unsigned int aclDepthMultiplier)
{
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;
}
arm_compute::experimental::dynamic_fusion::Pool2dAttributes
CreatePool2dAttributes(const Pooling2dDescriptor& descriptor)
{
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;
}
|
