repository

Contents

MaxPool

General

MaxPool operation performs the computation following the below formulas. Variable names follow the standard Naming Conventions.

\[\dst(n, c, oh, ow) = \max\limits_{kh, kw} \left( \src(n, c, oh \cdot SH + kh \cdot (DH + 1) - PH_L, ow \cdot SW + kw \cdot (DW + 1) - PW_L) \right)\]

Operation attributes

Attribute Name	Description	Value Type	Supported Values	Required or Optional
strides	Controls the strides the window is moved.	s64	A s64 list containing positive values	Required
pads_begin	Controls number of zeros to be add to the front/top/left of spatial dimensions, the attribute will be ignored when auto_pad attribute is specified to same_upper , same_lower or valid .	s64	A s64 list containing non-negative values	Required
pads_end	Controls number of zeros to be add to the back/bottom/right of spatial dimensions, the attribute will be ignored when auto_pad attribute is specified to same_upper , same_lower or valid .	s64	A s64 list containing non-negative values	Required
kernel	Size of pooling window.	s64	A s64 list containing positive values	Required
rounding_type	Controls how to do rounding.	string	floor (default), ceil	Optional
auto_pad	Controls how the paddings are calculated.	string	none (default), same_upper , same_lower , valid	Optional
dilations	Denotes the distance in width and height between elements in the window.	s64	A s64 list containing positive values, a list of 1 s (default) means no dilation	Optional
data_format	Controls how to interpret the shape of src and dst .	string	NCX , NXC (default)	Optional

Execution arguments

The inputs and outputs must be provided according to below index order when constructing an operation.

Inputs

Index	Argument Name	Required or Optional
0	src	Required

Outputs

Index	Argument Name	Required or Optional
0	dst	Required

Supported data types

MaxPool operation supports the following data type combinations.

Src	Dst
f32	f32
bf16	bf16
f16	f16

Maximum MaxPoolBackward