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

rename_conv2d_grad_grad (#52374)

上级 d972de56
...@@ -209,7 +209,7 @@ ...@@ -209,7 +209,7 @@
func : GeneralTernaryGradInferMeta func : GeneralTernaryGradInferMeta
param: [input, filter, grad_out] param: [input, filter, grad_out]
kernel : kernel :
func : conv2d_grad_grad func : conv2d_double_grad
optional : grad_input_grad, grad_filter_grad optional : grad_input_grad, grad_filter_grad
- backward_op : conv2d_transpose_double_grad - backward_op : conv2d_transpose_double_grad
......
...@@ -123,9 +123,12 @@ PD_REGISTER_KERNEL(depthwise_conv2d_grad, ...@@ -123,9 +123,12 @@ PD_REGISTER_KERNEL(depthwise_conv2d_grad,
PD_REGISTER_KERNEL( PD_REGISTER_KERNEL(
conv3d_grad, CPU, ALL_LAYOUT, phi::Conv3DGradKernel, float, double) {} conv3d_grad, CPU, ALL_LAYOUT, phi::Conv3DGradKernel, float, double) {}
PD_REGISTER_KERNEL( PD_REGISTER_KERNEL(conv2d_double_grad,
conv2d_grad_grad, CPU, ALL_LAYOUT, phi::ConvGradGradKernel, float, double) { CPU,
} ALL_LAYOUT,
phi::ConvGradGradKernel,
float,
double) {}
PD_REGISTER_KERNEL(conv3d_double_grad, PD_REGISTER_KERNEL(conv3d_double_grad,
CPU, CPU,
......
...@@ -55,6 +55,9 @@ PD_REGISTER_KERNEL( ...@@ -55,6 +55,9 @@ PD_REGISTER_KERNEL(
PD_REGISTER_KERNEL( PD_REGISTER_KERNEL(
conv3d_grad, GPU, ALL_LAYOUT, phi::Conv3DGradKernel, float, double) {} conv3d_grad, GPU, ALL_LAYOUT, phi::Conv3DGradKernel, float, double) {}
PD_REGISTER_KERNEL( PD_REGISTER_KERNEL(conv2d_double_grad,
conv2d_grad_grad, GPU, ALL_LAYOUT, phi::ConvGradGradKernel, float, double) { GPU,
} ALL_LAYOUT,
phi::ConvGradGradKernel,
float,
double) {}
...@@ -1473,7 +1473,7 @@ PD_REGISTER_KERNEL(depthwise_conv2d_grad, ...@@ -1473,7 +1473,7 @@ PD_REGISTER_KERNEL(depthwise_conv2d_grad,
phi::DepthwiseConvCudnnGradKernel, phi::DepthwiseConvCudnnGradKernel,
float, float,
phi::dtype::float16) {} phi::dtype::float16) {}
PD_REGISTER_KERNEL(conv2d_grad_grad, PD_REGISTER_KERNEL(conv2d_double_grad,
GPUDNN, GPUDNN,
ALL_LAYOUT, ALL_LAYOUT,
phi::ConvCudnnGradGradKernel, phi::ConvCudnnGradGradKernel,
...@@ -1512,7 +1512,7 @@ PD_REGISTER_KERNEL(conv3d_grad, ...@@ -1512,7 +1512,7 @@ PD_REGISTER_KERNEL(conv3d_grad,
double, double,
phi::dtype::float16, phi::dtype::float16,
phi::dtype::bfloat16) {} phi::dtype::bfloat16) {}
PD_REGISTER_KERNEL(conv2d_grad_grad, PD_REGISTER_KERNEL(conv2d_double_grad,
GPUDNN, GPUDNN,
ALL_LAYOUT, ALL_LAYOUT,
phi::ConvCudnnGradGradKernel, phi::ConvCudnnGradGradKernel,
...@@ -1555,7 +1555,7 @@ PD_REGISTER_KERNEL(conv3d_grad, ...@@ -1555,7 +1555,7 @@ PD_REGISTER_KERNEL(conv3d_grad,
double, double,
phi::dtype::float16) {} phi::dtype::float16) {}
PD_REGISTER_KERNEL(conv2d_grad_grad, PD_REGISTER_KERNEL(conv2d_double_grad,
GPUDNN, GPUDNN,
ALL_LAYOUT, ALL_LAYOUT,
phi::ConvCudnnGradGradKernel, phi::ConvCudnnGradGradKernel,
......
...@@ -42,7 +42,7 @@ KernelSignature Conv2dGradOpArgumentMapping(const ArgumentMappingContext& ctx) { ...@@ -42,7 +42,7 @@ KernelSignature Conv2dGradOpArgumentMapping(const ArgumentMappingContext& ctx) {
KernelSignature Conv2dDoubleGradOpArgumentMapping( KernelSignature Conv2dDoubleGradOpArgumentMapping(
const ArgumentMappingContext& ctx) { const ArgumentMappingContext& ctx) {
return KernelSignature("conv2d_grad_grad", return KernelSignature("conv2d_double_grad",
{"Input", "Filter", "DOutput", "DDInput", "DDFilter"}, {"Input", "Filter", "DOutput", "DDInput", "DDFilter"},
{"strides", {"strides",
"paddings", "paddings",
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册