update y layout (#70645)

liuruyan · web-flow · commit 24c1aa6936a4 · 2025-01-07T10:46:09.000+08:00
diff --git a/paddle/phi/infermeta/multiary.cc b/paddle/phi/infermeta/multiary.cc
@@ -973,6 +973,7 @@ void BatchNormInferMeta(const MetaTensor& x,
   }
   y->share_lod(x);
   y->set_dtype(x.dtype());
+  y->set_layout(x.layout());
 }
 
 void BatchNormInferInferMeta(const MetaTensor& x,

Original file line number	Diff line number	Diff line change
`@@ -973,6 +973,7 @@ void BatchNormInferMeta(const MetaTensor& x,`
`973`	`973`	`}`
`974`	`974`	`y->share_lod(x);`
`975`	`975`	`y->set_dtype(x.dtype());`
	`976`	`+ y->set_layout(x.layout());`
`976`	`977`	`}`
`977`	`978`
`978`	`979`	`void BatchNormInferInferMeta(const MetaTensor& x,`