diff --git a/paddle/phi/kernels/gpu/sigmoid_cross_entropy_with_logits_grad_kernel.cu b/paddle/phi/kernels/gpu/sigmoid_cross_entropy_with_logits_grad_kernel.cu index ae3cefd9e82d0c99742a5099a00923167fe7e4ce..598b0138fb3a11d63230b9b40071fe7972c48da5 100644 --- a/paddle/phi/kernels/gpu/sigmoid_cross_entropy_with_logits_grad_kernel.cu +++ b/paddle/phi/kernels/gpu/sigmoid_cross_entropy_with_logits_grad_kernel.cu @@ -89,7 +89,7 @@ void SigmoidCrossEntropyWithLogitsGradKernel(const Context &dev_ctx, reduce_dim.push_back(i); } - kernels::TensorReduceImpl>( + funcs::TensorReduceImpl>( dev_ctx, *counts_tensor, norm_tensor, diff --git a/paddle/phi/kernels/gpu/sigmoid_cross_entropy_with_logits_kernel.cu b/paddle/phi/kernels/gpu/sigmoid_cross_entropy_with_logits_kernel.cu index fb63badf56a75f9b8c0be3c7ae6f0c2455e11bd7..13d63f8d97e42cf55c4e2acc1d3f7bf92efef868 100644 --- a/paddle/phi/kernels/gpu/sigmoid_cross_entropy_with_logits_kernel.cu +++ b/paddle/phi/kernels/gpu/sigmoid_cross_entropy_with_logits_kernel.cu @@ -89,7 +89,7 @@ void SigmoidCrossEntropyWithLogitsKernel(const Context &dev_ctx, reduce_dim.push_back(i); } - kernels::TensorReduceImpl>( + funcs::TensorReduceImpl>( dev_ctx, *counts_tensor, norm_tensor,