AvgPoolBackward#
AvgPoolBackward operation accepts diffdst tensor and srcshape tensor (optional), and calculates diffsrc tensor.
Operation Attributes#
|
Description |
Value Type |
|
|
---|---|---|---|---|
Controls the strides the window is moved |
s64 |
A s64 list containing positive values |
Required |
|
Controls
number of
zeros to be
add to the
front/top/left
of spatial
dimensions,
the
attribute
will be
ignored
when
|
s64 |
A s64 list containing n on-negative values |
Required |
|
Controls
number of
zeros to be
add to the
back/bottom/right
of spatial
dimensions,
the
attribute
will be
ignored
when
|
s64 |
A s64 list containing n on-negative values |
Required |
|
Size of pooling window |
s64 |
A s64 list containing positive values |
Required |
|
Controls whether the padded values are counted |
bool |
True, False |
Required |
|
Controls how the paddings are calculated |
string |
|
Optional |
|
Controls
how to
interpret
the shape
of |
string |
|
Optional |
|
Denotes the shape of input of forward op |
string |
|
Optional |
Execution Arguments#
The inputs and outputs must be provided according to the below index order when constructing an operation.
Inputs#
Index |
Argument Name |
Required or Optional |
---|---|---|
0 |
|
Required |
1 |
|
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 |
|
Required |
Supported Data Types#
AvgPoolBackward operation supports the following data type combinations.
Diff_dst |
Diff_src |
Src_shape |
---|---|---|
f32 |
f32 |
s64 |
bf16 |
bf16 |
s64 |
f16 |
f16 |
s64 |