Contents

AvgPool

General

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

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

where,

  • when attribute exclude_pad is set to false, in which case \(DENOM = KH \cdot KW\),

  • when attribute exclude_pad is set to true, in which case \(DENOM\) equals to the size of overlap between an averaging window and images.

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

exclude_pad

Controls whether the padded values are counted.

bool

True, False

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

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

AvgPool operation supports the following data type combinations.

Src

Dst

f32

f32

bf16

bf16

f16

f16
Add AvgPoolBackward