未验证 提交 49bbd466 编写于 作者: Z zhangyuqin1998 提交者: GitHub

Rename conv2d transpose grad grad (#52371)

* Rename conv2d transpose grad grad

* fix
上级 262ea02a
...@@ -208,7 +208,7 @@ ...@@ -208,7 +208,7 @@
infer_meta : infer_meta :
func : Conv2dTransposeDoubleGradInferMeta func : Conv2dTransposeDoubleGradInferMeta
kernel : kernel :
func : conv2d_transpose_grad_grad func : conv2d_transpose_double_grad
- backward_op : conv2d_transpose_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) 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)
......
...@@ -136,7 +136,7 @@ PD_REGISTER_KERNEL(conv2d_transpose_grad, ...@@ -136,7 +136,7 @@ PD_REGISTER_KERNEL(conv2d_transpose_grad,
phi::Conv2dTransposeGradKernel, phi::Conv2dTransposeGradKernel,
float, float,
double) {} double) {}
PD_REGISTER_KERNEL(conv2d_transpose_grad_grad, PD_REGISTER_KERNEL(conv2d_transpose_double_grad,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::Conv2dTransposeDoubleGradKernel, phi::Conv2dTransposeDoubleGradKernel,
......
...@@ -1059,7 +1059,7 @@ PD_REGISTER_KERNEL(conv2d_transpose_grad, ...@@ -1059,7 +1059,7 @@ PD_REGISTER_KERNEL(conv2d_transpose_grad,
phi::Conv2dTransposeGradGPUDNNKernel, phi::Conv2dTransposeGradGPUDNNKernel,
float, float,
float16) {} float16) {}
PD_REGISTER_KERNEL(conv2d_transpose_grad_grad, PD_REGISTER_KERNEL(conv2d_transpose_double_grad,
GPUDNN, GPUDNN,
ALL_LAYOUT, ALL_LAYOUT,
phi::Conv2dTransposeDoubleGradGPUDNNKernel, phi::Conv2dTransposeDoubleGradGPUDNNKernel,
...@@ -1081,7 +1081,7 @@ PD_REGISTER_KERNEL(conv2d_transpose_grad, ...@@ -1081,7 +1081,7 @@ PD_REGISTER_KERNEL(conv2d_transpose_grad,
double, double,
float16, float16,
phi::dtype::bfloat16) {} phi::dtype::bfloat16) {}
PD_REGISTER_KERNEL(conv2d_transpose_grad_grad, PD_REGISTER_KERNEL(conv2d_transpose_double_grad,
GPUDNN, GPUDNN,
ALL_LAYOUT, ALL_LAYOUT,
phi::Conv2dTransposeDoubleGradGPUDNNKernel, phi::Conv2dTransposeDoubleGradGPUDNNKernel,
...@@ -1105,7 +1105,7 @@ PD_REGISTER_KERNEL(conv2d_transpose_grad, ...@@ -1105,7 +1105,7 @@ PD_REGISTER_KERNEL(conv2d_transpose_grad,
float, float,
double, double,
float16) {} float16) {}
PD_REGISTER_KERNEL(conv2d_transpose_grad_grad, PD_REGISTER_KERNEL(conv2d_transpose_double_grad,
GPUDNN, GPUDNN,
ALL_LAYOUT, ALL_LAYOUT,
phi::Conv2dTransposeDoubleGradGPUDNNKernel, phi::Conv2dTransposeDoubleGradGPUDNNKernel,
......
...@@ -48,7 +48,7 @@ KernelSignature Conv2dTransposeGradOpArgumentMapping( ...@@ -48,7 +48,7 @@ KernelSignature Conv2dTransposeGradOpArgumentMapping(
KernelSignature Conv2dTransposeDoubleGradOpArgumentMapping( KernelSignature Conv2dTransposeDoubleGradOpArgumentMapping(
const ArgumentMappingContext& ctx) { const ArgumentMappingContext& ctx) {
return KernelSignature("conv2d_transpose_grad_grad", return KernelSignature("conv2d_transpose_double_grad",
{"Input", "Filter", "DOutput", "DDInput", "DDFilter"}, {"Input", "Filter", "DOutput", "DDInput", "DDFilter"},
{"strides", {"strides",
"paddings", "paddings",
...@@ -123,6 +123,9 @@ KernelSignature DepthwiseConv2dTransposeGradOpArgumentMapping( ...@@ -123,6 +123,9 @@ KernelSignature DepthwiseConv2dTransposeGradOpArgumentMapping(
} // namespace phi } // namespace phi
PD_REGISTER_BASE_KERNEL_NAME(conv2d_transpose_grad_grad,
conv2d_transpose_double_grad);
PD_REGISTER_ARG_MAPPING_FN(conv2d_transpose, PD_REGISTER_ARG_MAPPING_FN(conv2d_transpose,
phi::Conv2dTransposeOpArgumentMapping); phi::Conv2dTransposeOpArgumentMapping);
PD_REGISTER_ARG_MAPPING_FN(conv2d_transpose_grad, PD_REGISTER_ARG_MAPPING_FN(conv2d_transpose_grad,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册