未验证 提交 f315489d 编写于 作者: X xiongkun 提交者: GitHub

Fix Einsum Infershape get None (#42493)

上级 35fadcdf
...@@ -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.
先完成此消息的编辑!
想要评论请 注册