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

Rename conv2d transpose grad grad (#52371)

* Rename conv2d transpose grad grad

* fix
上级 262ea02a
......@@ -208,7 +208,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)
......
......@@ -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,
......
......@@ -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,
......@@ -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,
......@@ -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,
......
......@@ -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",
......@@ -123,6 +123,9 @@ KernelSignature DepthwiseConv2dTransposeGradOpArgumentMapping(
} // namespace phi
PD_REGISTER_BASE_KERNEL_NAME(conv2d_transpose_grad_grad,
conv2d_transpose_double_grad);
PD_REGISTER_ARG_MAPPING_FN(conv2d_transpose,
phi::Conv2dTransposeOpArgumentMapping);
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.
先完成此消息的编辑!
想要评论请 注册