diff --git a/paddle/phi/kernels/onednn/clip_grad_kernel.cc b/paddle/phi/kernels/onednn/clip_grad_kernel.cc
index aded64616b1245592fb4393b58c85ef70624d523..03da47cfa65d363cc2f461024baa424adc30b3d8 100644
--- a/paddle/phi/kernels/onednn/clip_grad_kernel.cc
+++ b/paddle/phi/kernels/onednn/clip_grad_kernel.cc
@@ -48,7 +48,7 @@ void ClipGradKernel(const Context& dev_ctx,
 
 PD_REGISTER_KERNEL(clip_grad,
                    OneDNN,
-                   ALL_LAYOUT,
+                   ONEDNN,
                    phi::ClipGradKernel,
                    float,
                    phi::dtype::bfloat16) {}
diff --git a/paddle/phi/kernels/onednn/clip_kernel.cc b/paddle/phi/kernels/onednn/clip_kernel.cc
index 7538dd9708a9363e6f77f96d0af40d68a04cbc23..0accedb1724f29dcaaa939ef6ba73bfe8dec1666 100644
--- a/paddle/phi/kernels/onednn/clip_kernel.cc
+++ b/paddle/phi/kernels/onednn/clip_kernel.cc
@@ -43,4 +43,4 @@ void ClipKernel(const Context& dev_ctx,
 }  // namespace phi
 
 PD_REGISTER_KERNEL(
-    clip, OneDNN, ALL_LAYOUT, phi::ClipKernel, float, phi::dtype::bfloat16) {}
+    clip, OneDNN, ONEDNN, phi::ClipKernel, float, phi::dtype::bfloat16) {}
diff --git a/paddle/phi/kernels/onednn/concat_grad_kernel.cc b/paddle/phi/kernels/onednn/concat_grad_kernel.cc
index be962a96acaf7166d48490052e31ef1eca7d7272..c36331c66f6f7db2ed4b8735c42b3fa4f1d73585 100644
--- a/paddle/phi/kernels/onednn/concat_grad_kernel.cc
+++ b/paddle/phi/kernels/onednn/concat_grad_kernel.cc
@@ -78,7 +78,7 @@ void ConcatGradKernel(const Context& dev_ctx,
 
 PD_REGISTER_KERNEL(concat_grad,
                    OneDNN,
-                   ALL_LAYOUT,
+                   ONEDNN,
                    phi::ConcatGradKernel,
                    float,
                    phi::dtype::bfloat16) {}
diff --git a/paddle/phi/kernels/onednn/expand_grad_kernel.cc b/paddle/phi/kernels/onednn/expand_grad_kernel.cc
index dd8afdd84677ad4f8139047c306af60d5f160a8c..243505f990d823c37178528e90038d1d4eb3d2ea 100644
--- a/paddle/phi/kernels/onednn/expand_grad_kernel.cc
+++ b/paddle/phi/kernels/onednn/expand_grad_kernel.cc
@@ -86,7 +86,7 @@ void ExpandGradKernel(const Context& dev_ctx,
 
 PD_REGISTER_KERNEL(expand_grad,
                    OneDNN,
-                   ALL_LAYOUT,
+                   ONEDNN,
                    phi::ExpandGradKernel,
                    float,
                    phi::dtype::bfloat16) {}
diff --git a/paddle/phi/kernels/onednn/expand_kernel.cc b/paddle/phi/kernels/onednn/expand_kernel.cc
index 52d12bb100d0f8a03d1d1b645afc2b54bb77eff7..24aa0c401f5a8757b428fc0b7789bee5d58b910e 100644
--- a/paddle/phi/kernels/onednn/expand_kernel.cc
+++ b/paddle/phi/kernels/onednn/expand_kernel.cc
@@ -75,9 +75,5 @@ void ExpandKernel(const Context& dev_ctx,
 }
 }  // namespace phi
 
-PD_REGISTER_KERNEL(expand,
-                   OneDNN,
-                   ALL_LAYOUT,
-                   phi::ExpandKernel,
-                   float,
-                   phi::dtype::bfloat16) {}
+PD_REGISTER_KERNEL(
+    expand, OneDNN, ONEDNN, phi::ExpandKernel, float, phi::dtype::bfloat16) {}
diff --git a/paddle/phi/kernels/onednn/log_softmax_kernel.cc b/paddle/phi/kernels/onednn/log_softmax_kernel.cc
index 963bb7e0e3224d3de6ec7c4b6249b51ce2710e32..800d67a31cb5d75e158e2533ae96aaf9c6f64f60 100644
--- a/paddle/phi/kernels/onednn/log_softmax_kernel.cc
+++ b/paddle/phi/kernels/onednn/log_softmax_kernel.cc
@@ -65,7 +65,7 @@ void LogSoftmaxKernel(const Context& dev_ctx,
 
 PD_REGISTER_KERNEL(log_softmax,
                    OneDNN,
-                   ALL_LAYOUT,
+                   ONEDNN,
                    phi::LogSoftmaxKernel,
                    float,
                    phi::dtype::bfloat16) {}
diff --git a/paddle/phi/kernels/onednn/pad3d_kernel.cc b/paddle/phi/kernels/onednn/pad3d_kernel.cc
index 2d34e11afc4cde6de46ebafe28a14c2816dc8325..fae428b6abaed0b98524a4e92d95f0cb662a48d9 100644
--- a/paddle/phi/kernels/onednn/pad3d_kernel.cc
+++ b/paddle/phi/kernels/onednn/pad3d_kernel.cc
@@ -31,4 +31,4 @@ void Pad3dKernel(const Context& dev_ctx,
 }
 }  // namespace phi
 
-PD_REGISTER_KERNEL(pad3d, OneDNN, ALL_LAYOUT, phi::Pad3dKernel, float) {}
+PD_REGISTER_KERNEL(pad3d, OneDNN, ONEDNN, phi::Pad3dKernel, float) {}
diff --git a/paddle/phi/kernels/onednn/pad_kernel.cc b/paddle/phi/kernels/onednn/pad_kernel.cc
index 4177f000dba147caf98723090f2012ee4be9c747..96949da8bead55ad256ee9bfd61f5f8e62a6a568 100644
--- a/paddle/phi/kernels/onednn/pad_kernel.cc
+++ b/paddle/phi/kernels/onednn/pad_kernel.cc
@@ -34,4 +34,4 @@ void PadKernel(const Context& dev_ctx,
 }
 }  // namespace phi
 
-PD_REGISTER_KERNEL(pad, OneDNN, ALL_LAYOUT, phi::PadKernel, float) {}
+PD_REGISTER_KERNEL(pad, OneDNN, ONEDNN, phi::PadKernel, float) {}
diff --git a/paddle/phi/kernels/onednn/slice_grad_kernel.cc b/paddle/phi/kernels/onednn/slice_grad_kernel.cc
index c38a2237e54779fca524e4d4ddf9a5162c179442..45a152fd69413485bd05a28932bf745c7e07facc 100644
--- a/paddle/phi/kernels/onednn/slice_grad_kernel.cc
+++ b/paddle/phi/kernels/onednn/slice_grad_kernel.cc
@@ -80,7 +80,7 @@ void SliceGradRawKernel(const Context& dev_ctx,
 
 PD_REGISTER_KERNEL(slice_grad,
                    OneDNN,
-                   ALL_LAYOUT,
+                   ONEDNN,
                    phi::SliceGradRawKernel,
                    float,
                    phi::dtype::bfloat16) {}
diff --git a/paddle/phi/kernels/onednn/slice_kernel.cc b/paddle/phi/kernels/onednn/slice_kernel.cc
index 3f74a2fe0be01cba2234bff7dd9745384b8de909..252c464532ddf3236b2ff832bbbad4ef823e89f3 100644
--- a/paddle/phi/kernels/onednn/slice_kernel.cc
+++ b/paddle/phi/kernels/onednn/slice_kernel.cc
@@ -101,7 +101,7 @@ void SliceRawKernel(const Context& dev_ctx,
 
 PD_REGISTER_KERNEL(slice,
                    OneDNN,
-                   ALL_LAYOUT,
+                   ONEDNN,
                    phi::SliceRawKernel,
                    float,
                    int8_t,
diff --git a/paddle/phi/kernels/onednn/softmax_grad_kernel.cc b/paddle/phi/kernels/onednn/softmax_grad_kernel.cc
index 0209992a679cdc0c125b63c8e56f6747ad7bab3f..facbb9e9f193c0d6192ac933523232e532e09991 100644
--- a/paddle/phi/kernels/onednn/softmax_grad_kernel.cc
+++ b/paddle/phi/kernels/onednn/softmax_grad_kernel.cc
@@ -50,4 +50,4 @@ void SoftmaxGradKernel(const Context& dev_ctx,
 }  // namespace phi
 
 PD_REGISTER_KERNEL(
-    softmax_grad, OneDNN, ALL_LAYOUT, phi::SoftmaxGradKernel, float) {}
+    softmax_grad, OneDNN, ONEDNN, phi::SoftmaxGradKernel, float) {}