Update paddle/fluid/operators/mkldnn/test_mkldnn_op_nhwc.cc

Co-authored-by: Sławomir Siwek <[email protected]>
PaddlePaddle · Nov 3, 2022 · 19bd42b · 19bd42b
1 parent 16f2509
commit 19bd42b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/fluid/operators/mkldnn/test_mkldnn_op_nhwc.cc b/paddle/fluid/operators/mkldnn/test_mkldnn_op_nhwc.cc
@@ -28,7 +28,7 @@
 #include "paddle/phi/core/kernel_registry.h"
 
 USE_OP_ITSELF(pool2d);
-PD_DECLARE_KERNEL(pool2d, OneDNN, ALL_LAYOUT);
+PD_DECLARE_KERNEL(pool2d, OneDNN, ONEDNN);
 USE_OP_ITSELF(relu);
 PD_DECLARE_KERNEL(relu, OneDNN, ONEDNN);
 USE_OP_ITSELF(transpose);