ConvTransposeBackwardWeights¶
General¶
ConvTransposeBackwardWeights operation takes \(\diffdst\), \(\src\) and optional \(weights\_shape\) computes \(\diffweights\).
Operation attributes¶
Attribute Name |
Description |
Value Type |
Supported Values |
Required or Optional |
---|---|---|---|---|
Controls the strides the weights tensor is moved when computing convolution |
s64 |
A s64 list containing positive values |
Required |
|
Controls number of zeros to be add to the front/top/left of spatial dimensions |
s64 |
A s64 list containing non-negative values |
Required |
|
Controls number of zeros to be add to the back/bottom/right of spatial dimensions |
s64 |
A s64 list containing non-negative values |
Required |
|
Controls the amount of stretching the kernel before convolution ( visualization link ) |
s64 |
A s64 list containing positive values (>1 means dilated convolution) |
Required |
|
Controls how the padding is calculated |
string |
|
Optional |
|
Adds additional amount of padding per each spatial axis in |
s64 |
A s64 list containing non-negative values, all zeros by default |
Optional |
|
Controls how input channels and output channels are divided into |
s64 |
A positive s64 value, |
Optional |
|
Controls how to interpret the shape of |
string |
|
Optional |
|
Controls how to interpret the shape of |
string |
|
Optional |
|
Denotes the shape of the |
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 |
|
Required |
2 |
|
Optional |
Note
The shape of \(\weights\) is \((in\_channels / groups, out\_channels, spatial\_shape)\) for IOX
format or \((spatial\_shape, out\_channels, in\_channels / groups)\) for XOI
format. Both \(in\_channels\) and \(out\_channels\) must be divisible by groups attribute.
Note
Either weights_shape
input or weights_shape
attribute should be provided. If both provided, weights_shape
input will precede over weights_shape
attribute.
Outputs¶
Index |
Argument Name |
Required or Optional |
---|---|---|
0 |
|
Required |
Supported data types¶
ConvTransposeBackwardWeights operation supports the following data type combinations.
Src |
Diff_dst |
Diff_weights |
Weights_shape |
---|---|---|---|
f32 |
f32 |
f32 |
s32 |
bf16 |
bf16 |
bf16 |
s32 |
f16 |
f16 |
f16 |
s32 |