ConvTransposeBackwardWeights#

ConvTransposeBackwardWeights operation takes \(\diffdst\), src and optional \(weights\_shape\) computes \(\diffweights\).

Operation Attributes#

Attribute

Name

Description

Value Type

Supported

Values

Required or

Optional

strides

Controls the strides the weights tensor is moved when computing convolution

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

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

s64

A s64 list containing non-negative values

Required

dilations

Controls the amount of stretching the kernel before convolution

s64

A s64 list containing positive values (>1 means dilated convolution)

Required

auto_pad

Controls how the padding is calculated

string

none (default), same_upper, same_lower, valid

Optional

output_padding

Adds additional amount of padding per each spatial axis in dst

s64

A s64 list containing non-negative values, all zeros by default

Optional

groups

Controls how input channels and output channels are divided into

s64

A positive s64 value, 1 by default

Optional

data_format

Controls how to interpret the shape of src and dst.

string

NCX, NXC (default)

Optional

weights_format

Controls how to interpret the shape of weights

string

IOX, XOI (default)

Optional

weights_shape

Denotes the shape of the weights tensor

s64

A s64 list containing positive values

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

src

Required

1

diff_dst

Required

2

weights_shape

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 the weights_shape attribute.

Outputs#

Index

Argument Name

Required or Optional

0

diff_weights

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