24.04
DepthwiseConv2dAttributes.h
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2022 Arm Limited.
3
*
4
* SPDX-License-Identifier: MIT
5
*
6
* Permission is hereby granted, free of charge, to any person obtaining a copy
7
* of this software and associated documentation files (the "Software"), to
8
* deal in the Software without restriction, including without limitation the
9
* rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10
* sell copies of the Software, and to permit persons to whom the Software is
11
* furnished to do so, subject to the following conditions:
12
*
13
* The above copyright notice and this permission notice shall be included in all
14
* copies or substantial portions of the Software.
15
*
16
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22
* SOFTWARE.
23
*/
24
#ifndef ARM_COMPUTE_DYNAMIC_FUSION_SKETCH_ATTRIBUTES_DEPTHWISECONV2DATTRIBUTES
25
#define ARM_COMPUTE_DYNAMIC_FUSION_SKETCH_ATTRIBUTES_DEPTHWISECONV2DATTRIBUTES
26
27
#include "
arm_compute/core/Size2D.h
"
28
#include "
arm_compute/core/Types.h
"
29
30
#include <cstdint>
31
32
namespace
arm_compute
33
{
34
namespace
experimental
35
{
36
namespace
dynamic_fusion
37
{
38
/** Attributes are backend-agnostic parameters (in addition to the input/output tensors) of an operator.
39
*/
40
41
/** Depthwise Conv2d attributes */
42
class
DepthwiseConv2dAttributes
43
{
44
public
:
45
/** Set padding */
46
DepthwiseConv2dAttributes
&
pad
(
const
Padding2D
&
pad
);
47
/** Get padding */
48
Padding2D
pad
()
const
;
49
/** Set stride */
50
DepthwiseConv2dAttributes
&
stride
(
const
Size2D
&
stride
);
51
/** Get stride */
52
Size2D
stride
()
const
;
53
/** Set dilation */
54
DepthwiseConv2dAttributes
&
dilation
(
const
Size2D
&
dilation
);
55
/** Get dilation */
56
Size2D
dilation
()
const
;
57
/** Set depth multiplier */
58
DepthwiseConv2dAttributes
&
depth_multiplier
(
const
uint32_t &
depth_multiplier
);
59
/** Get depth multiplier */
60
uint32_t
depth_multiplier
()
const
;
61
/** Set Dimension rounding type */
62
DepthwiseConv2dAttributes
&
dimension_rounding_type
(
const
DimensionRoundingType
&
dimension_rounding_type
);
63
/** Get Dimension rounding type */
64
DimensionRoundingType
dimension_rounding_type
()
const
;
65
66
private
:
67
Padding2D
_pad{};
/**< Padding */
68
Size2D
_stride{1
U
, 1
U
};
/**< Stride */
69
Size2D
_dilation{1
U
, 1
U
};
/**< Dilation */
70
uint32_t _depth_multiplier{1
U
};
/**< Depth multiplier */
71
DimensionRoundingType
_dimension_rounding_type{
DimensionRoundingType::FLOOR
};
/**< Dimension rounding type */
72
};
73
}
// namespace dynamic_fusion
74
}
// namespace experimental
75
}
// namespace arm_compute
76
#endif
/* ARM_COMPUTE_DYNAMIC_FUSION_SKETCH_ATTRIBUTES_DEPTHWISECONV2DATTRIBUTES */
Types.h
arm_compute::Size2D
Class for specifying the size of an image or rectangle.
Definition:
Size2D.h:34
arm_compute::DimensionRoundingType
DimensionRoundingType
Dimension rounding type when down-scaling on CNNs.
Definition:
CoreTypes.h:133
arm_compute::utils::cast::U
U
Definition:
SaturateCast.h:65
arm_compute::experimental::dynamic_fusion::DepthwiseConv2dAttributes::stride
Size2D stride() const
Get stride.
Definition:
DepthwiseConv2dAttributes.cpp:47
Size2D.h
arm_compute::experimental::dynamic_fusion::DepthwiseConv2dAttributes::dilation
Size2D dilation() const
Get dilation.
Definition:
DepthwiseConv2dAttributes.cpp:56
arm_compute::experimental::dynamic_fusion::DepthwiseConv2dAttributes::pad
Padding2D pad() const
Get padding.
Definition:
DepthwiseConv2dAttributes.cpp:38
arm_compute::DimensionRoundingType::FLOOR
@ FLOOR
Floor rounding.
arm_compute::experimental::dynamic_fusion::DepthwiseConv2dAttributes::dimension_rounding_type
DimensionRoundingType dimension_rounding_type() const
Get Dimension rounding type.
Definition:
DepthwiseConv2dAttributes.cpp:79
arm_compute
Copyright (c) 2017-2024 Arm Limited.
Definition:
introduction.dox:24
arm_compute::experimental::dynamic_fusion::DepthwiseConv2dAttributes::depth_multiplier
uint32_t depth_multiplier() const
Get depth multiplier.
Definition:
DepthwiseConv2dAttributes.cpp:67
arm_compute::experimental::dynamic_fusion::DepthwiseConv2dAttributes
Attributes are backend-agnostic parameters (in addition to the input/output tensors) of an operator.
Definition:
DepthwiseConv2dAttributes.h:42
arm_compute::Padding2D
Padding and stride information class.
Definition:
Types.h:609
arm_compute
dynamic_fusion
sketch
attributes
DepthwiseConv2dAttributes.h
Generated on Mon Apr 29 2024 10:53:52 for Compute Library by
1.8.17