diff --git a/paddle/phi/api/yaml/legacy_backward.yaml b/paddle/phi/api/yaml/legacy_backward.yaml index 0232c03e2c3cdf5b6797dc209991bb8e1b1f032c..176491491b90c6c1f7c5bfaadfaf41d114b2c8ac 100755 --- a/paddle/phi/api/yaml/legacy_backward.yaml +++ b/paddle/phi/api/yaml/legacy_backward.yaml @@ -2460,7 +2460,7 @@ - backward_api : sync_batch_norm_grad forward : sync_batch_norm_ (Tensor x, Tensor scale, Tensor bias, Tensor mean, Tensor variance, float momentum, float epsilon, str data_layout, bool is_test, bool use_global_stats, bool trainable_statistics, bool fuse_with_relu) -> Tensor(out), Tensor(mean_out), Tensor(variance_out), Tensor(saved_mean), Tensor(saved_variance), Tensor(reserve_space) - args : (Tensor x, Tensor scale, Tensor bias, Tensor mean_out, Tensor variance_out, Tensor saved_mean, Tensor saved_variance, Tensor reserve_space, Tensor out_grad, float momentum, float epsilon, str data_layout, bool is_test, bool use_global_stats, bool trainable_statistics, bool fuse_with_relu) + args : (Tensor x, Tensor scale, Tensor bias, Tensor saved_mean, Tensor saved_variance, Tensor reserve_space, Tensor out_grad, float momentum, float epsilon, str data_layout, bool is_test, bool use_global_stats, bool trainable_statistics, bool fuse_with_relu) output : Tensor(x_grad), Tensor(scale_grad), Tensor(bias_grad) infer_meta : func : GeneralTernaryGradInferMeta @@ -2468,7 +2468,7 @@ kernel : func : sync_batch_norm_grad data_type : out_grad - optional : mean_out, variance_out, reserve_space + optional : reserve_space - backward_api : take_along_axis_grad forward : take_along_axis (Tensor x, Tensor index, int axis) -> Tensor(out) diff --git a/paddle/phi/kernels/gpu/sync_batch_norm_grad_kernel.cu b/paddle/phi/kernels/gpu/sync_batch_norm_grad_kernel.cu index ba5020d08bd0f6953eb5ba31e1dd1f5287c0435c..5cf7aabda303a73b581ee498fb4ed5f42477af86 100644 --- a/paddle/phi/kernels/gpu/sync_batch_norm_grad_kernel.cu +++ b/paddle/phi/kernels/gpu/sync_batch_norm_grad_kernel.cu @@ -24,8 +24,6 @@ void SyncBatchNormGradKernel(const Context& ctx, const DenseTensor& x, const DenseTensor& scale, const DenseTensor& bias, - const paddle::optional& mean, - const paddle::optional& variance, const DenseTensor& saved_mean, const DenseTensor& saved_variance, const paddle::optional& reserve_space, diff --git a/paddle/phi/kernels/sync_batch_norm_grad_kernel.h b/paddle/phi/kernels/sync_batch_norm_grad_kernel.h index 395bec23f1091a4f24bf9d20dd7975118f253056..a38f42c29f62d090cce9debb54543043b26f51ad 100644 --- a/paddle/phi/kernels/sync_batch_norm_grad_kernel.h +++ b/paddle/phi/kernels/sync_batch_norm_grad_kernel.h @@ -25,8 +25,6 @@ void SyncBatchNormGradKernel(const Context& dev_ctx, const DenseTensor& x, const DenseTensor& scale, const DenseTensor& bias, - const paddle::optional& mean, - const paddle::optional& variance, const DenseTensor& saved_mean, const DenseTensor& saved_variance, const paddle::optional& reserve_space, diff --git a/paddle/phi/ops/compat/sync_batch_norm_sig.cc b/paddle/phi/ops/compat/sync_batch_norm_sig.cc index 2595f241ff2335ea419cd2096cfefef570cedf1c..9380751987ebc4865ef98ee7837a38adccfaa133 100644 --- a/paddle/phi/ops/compat/sync_batch_norm_sig.cc +++ b/paddle/phi/ops/compat/sync_batch_norm_sig.cc @@ -42,8 +42,6 @@ KernelSignature SyncBatchNormGradOpArgumentMapping( "X", "Scale", "Bias", - "Mean", - "Variance", "SavedMean", "SavedVariance", "ReserveSpace",