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

rename_batch_norm_grad_grad (#52372)

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