提交 267d6127 编写于 作者: M Megvii Engine Team

fix(imperative/mini_graph): fix no_err value in shape infer

GitOrigin-RevId: 087d15db661b2ea366a007eccfd15bd94ae58d79
上级 bc3561c1
......@@ -606,6 +606,7 @@ public:
desc.comp_node = minigraph.output_var(i)->comp_node();
if (auto* shape = sess.infer_shape(i, false)) {
desc.layout.init_contiguous_stride(*shape);
noerr = true;
} else {
noerr = false;
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册