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

rename_conv2d_grad_grad (#52374)

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