AvgPoolBackward

General

AvgPoolBackward operation accepts \(\diffdst\) tensor and \(\srcshape\) tensor (optional), and calculates \(\diffsrc\) tensor.

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

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

src_shape

Denotes the shape of input of forward op.

s64

A s64 list containing positive values.

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

diff_dst

Required

1

src_shape

Optional

Note

Either src_shape input or src_shape attribute should be provided. If both provided, src_shape input will precede over src_shape attribute.

Outputs

Index

Argument Name

Required or Optional

0

diff_src

Required

Supported data types

AvgPoolBackward operation supports the following data type combinations.

Diff_dst

Diff_src

Src_shape

f32

f32

s32

bf16

bf16

s32

f16

f16

s32