Skip to content

Commit

Permalink
Rename conv2d transpose grad grad
Browse files Browse the repository at this point in the history
  • Loading branch information
zhangyuqin1998 committed Apr 1, 2023
1 parent 8ef9708 commit e49e1c2
Show file tree
Hide file tree
Showing 4 changed files with 6 additions and 6 deletions.
2 changes: 1 addition & 1 deletion paddle/phi/api/yaml/legacy_backward.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -219,7 +219,7 @@
infer_meta :
func : Conv2dTransposeDoubleGradInferMeta
kernel :
func : conv2d_transpose_grad_grad
func : conv2d_transpose_double_grad

- backward_op : conv2d_transpose_grad
forward : conv2d_transpose(Tensor x, Tensor filter, int[] strides, int[] paddings, int[] output_padding, IntArray output_size, str padding_algorithm, int groups, int[] dilations, str data_format) -> Tensor(out)
Expand Down
2 changes: 1 addition & 1 deletion paddle/phi/kernels/gpu/conv_transpose_grad_kernel.cu
Original file line number Diff line number Diff line change
Expand Up @@ -136,7 +136,7 @@ PD_REGISTER_KERNEL(conv2d_transpose_grad,
phi::Conv2dTransposeGradKernel,
float,
double) {}
PD_REGISTER_KERNEL(conv2d_transpose_grad_grad,
PD_REGISTER_KERNEL(conv2d_transpose_double_grad,
GPU,
ALL_LAYOUT,
phi::Conv2dTransposeDoubleGradKernel,
Expand Down
6 changes: 3 additions & 3 deletions paddle/phi/kernels/gpudnn/conv_transpose_grad_kernel.cu
Original file line number Diff line number Diff line change
Expand Up @@ -1059,7 +1059,7 @@ PD_REGISTER_KERNEL(conv2d_transpose_grad,
phi::Conv2dTransposeGradGPUDNNKernel,
float,
float16) {}
PD_REGISTER_KERNEL(conv2d_transpose_grad_grad,
PD_REGISTER_KERNEL(conv2d_transpose_double_grad,
GPUDNN,
ALL_LAYOUT,
phi::Conv2dTransposeDoubleGradGPUDNNKernel,
Expand All @@ -1081,7 +1081,7 @@ PD_REGISTER_KERNEL(conv2d_transpose_grad,
double,
float16,
phi::dtype::bfloat16) {}
PD_REGISTER_KERNEL(conv2d_transpose_grad_grad,
PD_REGISTER_KERNEL(conv2d_transpose_double_grad,
GPUDNN,
ALL_LAYOUT,
phi::Conv2dTransposeDoubleGradGPUDNNKernel,
Expand All @@ -1105,7 +1105,7 @@ PD_REGISTER_KERNEL(conv2d_transpose_grad,
float,
double,
float16) {}
PD_REGISTER_KERNEL(conv2d_transpose_grad_grad,
PD_REGISTER_KERNEL(conv2d_transpose_double_grad,
GPUDNN,
ALL_LAYOUT,
phi::Conv2dTransposeDoubleGradGPUDNNKernel,
Expand Down
2 changes: 1 addition & 1 deletion paddle/phi/ops/compat/conv_transpose_sig.cc
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,7 @@ KernelSignature Conv2dTransposeGradOpArgumentMapping(

KernelSignature Conv2dTransposeDoubleGradOpArgumentMapping(
const ArgumentMappingContext& ctx) {
return KernelSignature("conv2d_transpose_grad_grad",
return KernelSignature("conv2d_transpose_double_grad",
{"Input", "Filter", "DOutput", "DDInput", "DDFilter"},
{"strides",
"paddings",
Expand Down

0 comments on commit e49e1c2

Please sign in to comment.