@@ -3184,11 +3184,11 @@ void FillSplitOutDims(const MetaTensor& x,
3184
3184
(*out)[i]->set_dtype (x.dtype ());
3185
3185
(*out)[i]->set_dims (out_dims[i]);
3186
3186
(*out)[i]->set_layout (x.layout ());
3187
+ (*out)[i]->share_lod (x);
3187
3188
} else {
3188
3189
(*out)[i]->set_dtype (x.dtype ());
3189
3190
(*out)[i]->set_dims (out_dims[i]);
3190
3191
(*out)[i]->set_layout (x.layout ());
3191
- (*out)[i]->share_lod (x);
3192
3192
}
3193
3193
}
3194
3194
}
@@ -3219,11 +3219,11 @@ void SplitInferMeta(const MetaTensor& x,
3219
3219
out[i]->set_dtype (x.dtype ());
3220
3220
out[i]->set_dims (out_dims[i]);
3221
3221
out[i]->set_layout (x.layout ());
3222
+ out[i]->share_lod (x);
3222
3223
} else {
3223
3224
out[i]->set_dtype (x.dtype ());
3224
3225
out[i]->set_dims (out_dims[i]);
3225
3226
out[i]->set_layout (x.layout ());
3226
- out[i]->share_lod (x);
3227
3227
}
3228
3228
}
3229
3229
} else {
@@ -3310,11 +3310,11 @@ void SplitWithNumInferMeta(const MetaTensor& x,
3310
3310
out[i]->set_dtype (x.dtype ());
3311
3311
out[i]->set_dims (out_dims[i]);
3312
3312
out[i]->set_layout (x.layout ());
3313
+ out[i]->share_lod (x);
3313
3314
} else {
3314
3315
out[i]->set_dtype (x.dtype ());
3315
3316
out[i]->set_dims (out_dims[i]);
3316
3317
out[i]->set_layout (x.layout ());
3317
- out[i]->share_lod (x);
3318
3318
}
3319
3319
}
3320
3320
} else {
0 commit comments