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 |
---|---|---|---|---|
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 non-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 non-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. |
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 |
|
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 |
s32 |
bf16 |
bf16 |
s32 |
f16 |
f16 |
s32 |