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

rename_batch_norm_grad_grad (#52372)

上级 eddf1ad6
......@@ -115,7 +115,7 @@
func : GeneralTernaryGradInferMeta
param : [x, scale, x]
kernel :
func : batch_norm_grad_grad
func : batch_norm_double_grad
data_type : x
optional : out_mean, out_variance, grad_x_grad, grad_scale_grad, grad_bias_grad
inplace : (grad_out -> grad_out_grad)
......
......@@ -662,7 +662,7 @@ PD_REGISTER_KERNEL(batch_norm_grad_raw,
float,
double) {}
PD_REGISTER_KERNEL(batch_norm_grad_grad,
PD_REGISTER_KERNEL(batch_norm_double_grad,
CPU,
ALL_LAYOUT,
phi::BatchNormDoubleGradKernel,
......
......@@ -1345,14 +1345,14 @@ PD_REGISTER_KERNEL(batch_norm_grad_raw,
#endif
#ifdef PADDLE_WITH_HIP
PD_REGISTER_KERNEL(batch_norm_grad_grad,
PD_REGISTER_KERNEL(batch_norm_double_grad,
GPU,
ALL_LAYOUT,
phi::BatchNormDoubleGradKernel,
float,
double) {}
#else
PD_REGISTER_KERNEL(batch_norm_grad_grad,
PD_REGISTER_KERNEL(batch_norm_double_grad,
GPU,
ALL_LAYOUT,
phi::BatchNormDoubleGradKernel,
......
......@@ -79,7 +79,7 @@ KernelSignature BatchNormGradOpArgumentMapping(
KernelSignature BatchNormGradGradOpArgumentMapping(
const ArgumentMappingContext& ctx) {
return KernelSignature("batch_norm_grad_grad",
return KernelSignature("batch_norm_double_grad",
{"X",
"Scale",
"Mean",
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册