File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -3789,7 +3789,7 @@ def conv2d_backward_weight(
37893789 r"""The gradient of conv2d with respect to weight.
37903790
37913791 This operator takes the output gradient `grad` as the convolution kernel
3792- and convolves it with `data` to produce the gradeint with respect to weight.
3792+ and convolves it with `data` to produce the gradient with respect to weight.
37933793 Depending on an implementation, the roles of `data` and `grad` can be swapped
37943794 (For example, in CUTLASS `data` acts as the filter).
37953795
Original file line number Diff line number Diff line change @@ -651,7 +651,7 @@ RELAY_REGISTER_OP("nn.conv2d_backward_weight")
651651 .describe(R"code( The gradient of the 2D convolution layer with respect to the weight.
652652
653653This layer computes the gradient of the conv2d op with respect to weight,
654- given the origial input data and the output gradient.
654+ given the original input data and the output gradient.
655655
656656- **data**: This depends on the `layout` parameter. Input is 4D array of shape
657657 (batch_size, in_channels, height, width) if `layout` is `NCHW`.
You can’t perform that action at this time.
0 commit comments