diff --git a/paddle/phi/kernels/onednn/add_n_kernel.cc b/paddle/phi/kernels/onednn/add_n_kernel.cc
index 421e60504be950f618e5f663c7328a190a35f353..413408a21e00052bd2cd1dda72e64e1fd66397bb 100644
--- a/paddle/phi/kernels/onednn/add_n_kernel.cc
+++ b/paddle/phi/kernels/onednn/add_n_kernel.cc
@@ -132,4 +132,4 @@ void AddNKernel(const Context& dev_ctx,
 }  // namespace phi
 
 PD_REGISTER_KERNEL(
-    add_n, OneDNN, ALL_LAYOUT, phi::AddNKernel, float, phi::dtype::bfloat16) {}
+    add_n, OneDNN, ONEDNN, phi::AddNKernel, float, phi::dtype::bfloat16) {}
diff --git a/paddle/phi/kernels/onednn/cast_kernel.cc b/paddle/phi/kernels/onednn/cast_kernel.cc
index 166db43db665dca68bc235fb5fd069eb92966038..74298cc055e0cb0758fb9622ccef34c3456b2cf9 100644
--- a/paddle/phi/kernels/onednn/cast_kernel.cc
+++ b/paddle/phi/kernels/onednn/cast_kernel.cc
@@ -56,4 +56,4 @@ void CastKernel(const Context& dev_ctx,
 }  // namespace phi
 
 PD_REGISTER_KERNEL(
-    cast, OneDNN, ALL_LAYOUT, phi::CastKernel, float, phi::dtype::bfloat16) {}
+    cast, OneDNN, ONEDNN, phi::CastKernel, float, phi::dtype::bfloat16) {}
diff --git a/paddle/phi/kernels/onednn/full_kernel.cc b/paddle/phi/kernels/onednn/full_kernel.cc
index 5a444175bfb4920564613200a07e8d94d60908b9..2820427af4d3cf3a3b883b93d8060c12f00bca13 100644
--- a/paddle/phi/kernels/onednn/full_kernel.cc
+++ b/paddle/phi/kernels/onednn/full_kernel.cc
@@ -85,4 +85,4 @@ void FullKernel(const Context& dev_ctx,
 }
 }  // namespace phi
 
-PD_REGISTER_KERNEL(full, OneDNN, ALL_LAYOUT, phi::FullKernel, float) {}
+PD_REGISTER_KERNEL(full, OneDNN, ONEDNN, phi::FullKernel, float) {}
diff --git a/paddle/phi/kernels/onednn/gaussian_random_kernel.cc b/paddle/phi/kernels/onednn/gaussian_random_kernel.cc
index b8259754d0b84eb035487c5412cf1a58b1d9834a..853cbfaa8a3c89743c78b95df972e8274c97a283 100644
--- a/paddle/phi/kernels/onednn/gaussian_random_kernel.cc
+++ b/paddle/phi/kernels/onednn/gaussian_random_kernel.cc
@@ -47,4 +47,4 @@ void GaussianRandomKernel(const Context& ctx,
 }  // namespace phi
 
 PD_REGISTER_KERNEL(
-    gaussian_random, OneDNN, ALL_LAYOUT, phi::GaussianRandomKernel, float) {}
+    gaussian_random, OneDNN, ONEDNN, phi::GaussianRandomKernel, float) {}
diff --git a/paddle/phi/kernels/onednn/interpolate_kernel.cc b/paddle/phi/kernels/onednn/interpolate_kernel.cc
index f70b9bcaf1a6cd2d98c022e415d4a6167824d976..abef0ccf6feb31da6fb854a17356e9cb2f793332 100644
--- a/paddle/phi/kernels/onednn/interpolate_kernel.cc
+++ b/paddle/phi/kernels/onednn/interpolate_kernel.cc
@@ -228,11 +228,11 @@ void NearestInterpKernel(
 }  // namespace phi
 
 PD_REGISTER_KERNEL(
-    bilinear_interp, OneDNN, ALL_LAYOUT, phi::BilinearInterpKernel, float) {}
+    bilinear_interp, OneDNN, ONEDNN, phi::BilinearInterpKernel, float) {}
 
 PD_REGISTER_KERNEL(nearest_interp,
                    OneDNN,
-                   ALL_LAYOUT,
+                   ONEDNN,
                    phi::NearestInterpKernel,
                    float,
                    phi::dtype::bfloat16,
diff --git a/paddle/phi/kernels/onednn/scale_kernel.cc b/paddle/phi/kernels/onednn/scale_kernel.cc
index 9ff767cff8ca31c11007914aa2a167f6b3678d24..68bee7a39c8a5615bc8c77072f38f1e352d6df8f 100644
--- a/paddle/phi/kernels/onednn/scale_kernel.cc
+++ b/paddle/phi/kernels/onednn/scale_kernel.cc
@@ -59,4 +59,4 @@ void ScaleKernel(const Context& dev_ctx,
 }  // namespace phi
 
 PD_REGISTER_KERNEL(
-    scale, OneDNN, ALL_LAYOUT, phi::ScaleKernel, float, phi::dtype::bfloat16) {}
+    scale, OneDNN, ONEDNN, phi::ScaleKernel, float, phi::dtype::bfloat16) {}
diff --git a/paddle/phi/kernels/onednn/sgd_kernel.cc b/paddle/phi/kernels/onednn/sgd_kernel.cc
index bbb02204105d16afd40acbd58003062f246f0fa3..5bdcbf03e1f5bde2b6f6445899e774c4f59059a5 100644
--- a/paddle/phi/kernels/onednn/sgd_kernel.cc
+++ b/paddle/phi/kernels/onednn/sgd_kernel.cc
@@ -82,12 +82,11 @@ void SGDDenseParamSparseGradKernel(
 }  // namespace phi
 
 PD_REGISTER_KERNEL(
-    sgd, OneDNN, ALL_LAYOUT, phi::SGDDenseKernel, float, phi::dtype::bfloat16) {
-}
+    sgd, OneDNN, ONEDNN, phi::SGDDenseKernel, float, phi::dtype::bfloat16) {}
 
 PD_REGISTER_KERNEL(sgd_dense_param_sparse_grad,
                    OneDNN,
-                   ALL_LAYOUT,
+                   ONEDNN,
                    phi::SGDDenseParamSparseGradKernel,
                    float,
                    phi::dtype::bfloat16) {}
diff --git a/paddle/phi/kernels/onednn/split_kernel.cc b/paddle/phi/kernels/onednn/split_kernel.cc
index 1d0544758fe96f3b830a1a4b394797b6e25e28d5..94c0924e36e00bf60c5e0fbc4287c2321b458b80 100644
--- a/paddle/phi/kernels/onednn/split_kernel.cc
+++ b/paddle/phi/kernels/onednn/split_kernel.cc
@@ -80,11 +80,11 @@ void SplitWithNumKernel(const Context& dev_ctx,
 }  // namespace phi
 
 PD_REGISTER_KERNEL(
-    split, OneDNN, ALL_LAYOUT, phi::SplitKernel, float, phi::dtype::bfloat16) {}
+    split, OneDNN, ONEDNN, phi::SplitKernel, float, phi::dtype::bfloat16) {}
 
 PD_REGISTER_KERNEL(split_with_num,
                    OneDNN,
-                   ALL_LAYOUT,
+                   ONEDNN,
                    phi::SplitWithNumKernel,
                    float,
                    phi::dtype::bfloat16) {}
diff --git a/paddle/phi/kernels/onednn/stack_kernel.cc b/paddle/phi/kernels/onednn/stack_kernel.cc
index 6ede31952e88d1caa3e3ca32ec6806151d74a4dd..ba418f4de6fcbabfc96f9ec338445af57a509e99 100644
--- a/paddle/phi/kernels/onednn/stack_kernel.cc
+++ b/paddle/phi/kernels/onednn/stack_kernel.cc
@@ -124,4 +124,4 @@ void StackKernel(const Context& dev_ctx,
 
 }  // namespace phi
 
-PD_REGISTER_KERNEL(stack, OneDNN, ALL_LAYOUT, phi::StackKernel, float) {}
+PD_REGISTER_KERNEL(stack, OneDNN, ONEDNN, phi::StackKernel, float) {}