Skip to content

P
Paddle

PaddlePaddle / Paddle
大约 1 年 前同步成功

通知 2299
Star 20931
Fork 5422
P
Paddle
未验证 提交 f315489d 编写于 作者: X xiongkun 提交者： GitHub
浏览文件
操作

Fix Einsum Infershape get None (#42493)

上级 35fadcdf
显示空白变更内容
内联 并排
Showing with 2 addition and 0 deletion
paddle/phi/infermeta/unary.cc
浏览文件 @ f315489d
...@@ -436,6 +436,8 @@ void EinsumInferMeta(const std::vector<const MetaTensor*>& inputs, ...@@ -436,6 +436,8 @@ void EinsumInferMeta(const std::vector<const MetaTensor*>& inputs,
<< paddle::string::join_strings(output_dims, ","); << paddle::string::join_strings(output_dims, ",");
VLOG(3) << "Label Type is : " << label_to_string(all_labels, labeltype); VLOG(3) << "Label Type is : " << label_to_string(all_labels, labeltype);
VLOG(3) << "Label Shape is : " << label_to_string(all_labels, labelshape); VLOG(3) << "Label Shape is : " << label_to_string(all_labels, labelshape);
out->set_dims(make_ddim(output_dims));
out->set_dtype(inputs[0]->dtype());
} }
void ExpandInferMeta(const MetaTensor& x, void ExpandInferMeta(const MetaTensor& x,
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或