diff --git a/paddle/phi/infermeta/backward.cc b/paddle/phi/infermeta/backward.cc index 1eca092a5f22f0fdc15b97eaadb288e456bfdda3..a33b9587c153ca2f7d67966a3a1a3838ebece5ee 100644 --- a/paddle/phi/infermeta/backward.cc +++ b/paddle/phi/infermeta/backward.cc @@ -419,6 +419,7 @@ void InverseGradInferMeta(const MetaTensor& out, MetaTensor* dx) { if (dx) { dx->set_dims(dout.dims()); + dx->set_dtype(out.dtype()); } } diff --git a/paddle/phi/infermeta/unary.cc b/paddle/phi/infermeta/unary.cc index edc455225e4dc272cf88cd77549149a44766de00..c018e58a59a37b537fe55f11ce346e516b0390a9 100644 --- a/paddle/phi/infermeta/unary.cc +++ b/paddle/phi/infermeta/unary.cc @@ -1059,6 +1059,7 @@ void InverseInferMeta(const MetaTensor& x, MetaTensor* out) { } out->set_dims(input_dims); + out->set_dtype(x.dtype()); out->share_lod(x); }