未验证 提交 ce7c7995 编写于 作者: F freeliuzc 提交者: GitHub

add set_dtype for inverse_op (#44618)

上级 a71cfd8c
......@@ -419,6 +419,7 @@ void InverseGradInferMeta(const MetaTensor& out,
MetaTensor* dx) {
if (dx) {
dx->set_dims(dout.dims());
dx->set_dtype(out.dtype());
}
}
......
......@@ -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);
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册