Skip to content

Commit fc0ef66

Browse files
authored
fix bug of layout convert (#71207)
1 parent 26b648d commit fc0ef66

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

paddle/fluid/pir/transforms/general/auto_layout_insert_pass.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -342,6 +342,7 @@ const std::set<std::string> ops_in_NCHW = {"pd_op.max_pool2d_with_index",
342342
// "pd_op.batch_norm_",
343343
"pd_op.bilinear_interp",
344344
"pd_op.shape",
345+
"pd_op.shape64",
345346
"pd_op.deformable_conv",
346347
"pd_op.set_value_with_tensor_",
347348
"pd_op.set_value_with_tensor"};

0 commit comments

Comments
 (0)